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()
量子化エンベディング。

継承されるメソッド

ブール値
equals(Object arg0)
final クラス<?>
getClass()
整数
hashCode()
final void
notify()
final void
notifyAll()
文字列
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

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

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 ()

量子化エンベディング。

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