Embedding

公共抽象类 Embedding

表示给定嵌入器头部的嵌入。通常用于嵌入任务。

“floatEmbedding”和“quantizedEmbedding”中的一个(且只有一个)包含数据,具体取决于嵌入器是否配置为执行 Scala 量化。

公共构造函数

公共方法

静态 嵌入
create(float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, Optional<String> headName)
创建 Embedding 实例。
静态 嵌入
createFromProto(EmbeddingsProto.Embedding proto)
根据 ERROR(/EmbeddingsProto.Embedding) protobuf 消息创建 Embedding 对象。
abstract float[]
floatEmbedding()
浮点嵌入。
abstract int
headIndex()
这些条目所引用的嵌入器头的索引。
abstract 可选<String>
headName()
嵌入器头部的可选名称,即相应的张量元数据名称。
abstract byte[]
quantizedEmbedding()
量化嵌入。

继承的方法

公共构造函数

public 嵌入 ()

公共方法

public static Embedding create (float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, 可选<String> headName)

创建 Embedding 实例。

参数
floatEmbedding 浮点嵌入
quantizedEmbedding 量化嵌入。
headIndex 嵌入器头部的索引。
headName 嵌入器头部的可选名称。

public static Embedding createFromProto (EmbeddingsProto.Embedding proto)

根据 ERROR(/EmbeddingsProto.Embedding) protobuf 消息创建 Embedding 对象。

参数
proto 要转换的 ERROR(/EmbeddingsProto.Embedding) protobuf 消息。

public abstract float[] floatEmbedding ()

浮点嵌入。

如果嵌入器已配置为执行标量量化,则为空。

public abstract int headIndex ()

这些条目所引用的嵌入器头的索引。这对于多头模型非常有用。

public abstract 可选<String> headName ()

嵌入器头部的可选名称,即相应的张量元数据名称。

public abstract byte[] quantizedEmbedding ()

量化嵌入。

如果未将嵌入器配置为执行标量量化,则为空。