Embedding

public 抽象クラス Embedding

特定のエンベダー ヘッドのエンベディングを表します。通常、エンベディング タスクで使用されます。

「floatEmbedding」と「quantizedEmbedding」の 2 つのうちの 1 つだけには、エンベダがスカラー量子化を実行するように構成されているかどうかに基づいて、データが含まれます。

パブリック コンストラクタ

パブリック メソッド

静的埋め込み
create(float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, Optional<String> headName)
Embedding インスタンスを作成します。
静的埋め込み
createFromProto(EmbeddingsProto.Embedding プロトコル)
ERROR(/EmbeddingsProto.Embedding) protobuf メッセージから Embedding オブジェクトを作成します。
抽象 float[]
floatEmbedding()
浮動小数点埋め込み。
抽象 整数
headIndex()
これらのエントリが参照するエンベダー ヘッドのインデックス。
抽象 省略可<文字列>
headName()
エンベダー ヘッドの任意の名前。対応するテンソル メタデータ名です。
抽象 byte[]
quantizedEmbedding()
量子化エンベディング。

継承されるメソッド

パブリック コンストラクタ

public エンベディング ()

パブリック メソッド

public static Embedding create (float[] floatEmbedding, byte[] quantizedEmbedding, int headIndex, 省略可<文字列> headName)

Embedding インスタンスを作成します。

パラメータ
floatEmbedding 浮動小数点エンベディング
quantizedEmbedding 量子化エンベディングです。
headIndex 埋め込みヘッドのインデックス。
headName エンベダー ヘッドの名前(省略可)。

public static Embedding createFromProto (EmbeddingsProto.Embedding proto)

ERROR(/EmbeddingsProto.Embedding) protobuf メッセージから Embedding オブジェクトを作成します。

パラメータ
proto 変換する ERROR(/EmbeddingsProto.Embedding) protobuf メッセージ。

public 抽象 float[] floatEmbedding ()

浮動小数点埋め込み。

エンベダーがスカラー量子化を実行するように構成されている場合は空になります。

public 抽象 int headIndex ()

これらのエントリが参照するエンベダー ヘッドのインデックス。これはマルチヘッド モデルの場合に便利です。

public 抽象 省略可<String> headName ()

エンベダー ヘッドの任意の名前。対応するテンソル メタデータ名です。

public 抽象 byte[] quantizedEmbedding ()

量子化エンベディング。

エンベダーがスカラー量子化を実行するように構成されていない場合は空になります。