TextEmbedder

テキストに対してエンベディング抽出を実行します。

この API は、TFLite モデル メタデータ（オプション）を持つ TFLite モデルを想定しています。

int32 入力テンソルを使用するモデルには、モデルのトークナイザの入力プロセス単位が含まれているため、メタデータが必要です。文字列入力テンソルを使用するモデルにメタデータは必要ありません。

• 入力テンソル
  • 入力 ID、マスク ID、セグメント ID を表す、形状 [batch_size x bert_max_seq_len] の 3 つの入力テンソル（kTfLiteInt32）。この入力署名には、モデル メタデータに BERT Tokenizer プロセス ユニットが必要です。
  • または、入力 ID を表す形 [batch_size x max_seq_len] の 1 つの入力テンソル（kTfLiteInt32）。この入力署名には、モデル メタデータに Regex Tokenizer プロセス ユニットが必要です。
  • または、シェイプレス、または入力文字列を含む形状 [1] の 1 つの入力テンソル（kTfLiteString）。
• 形状 [1 x N] の少なくとも 1 つの出力テンソル（kTfLiteFloat32/kTfLiteUint8）。ここで、N は生成されたエンベディングの次元数です。