TextEmbedder

執行文字的嵌入擷取作業。

這個 API 預期的 TFLite 模型為 (選用) TFLite 模型中繼資料。

含有 int32 輸入張量的模型必須具備中繼資料，因為其中含有模型權杖化工具的輸入程序單位。具備字串輸入張量的模型不需要中繼資料。

• 輸入張量
  • 形狀 [batch_size x bert_max_seq_len] 的三個輸入張量 (kTfLiteInt32)，代表輸入 ID、遮罩 ID 和區隔 ID。這項輸入簽章需要模型中繼資料中的 BERT 權杖化工具程序單元。
  • 或是形狀 [batch_size x max_seq_len] 的一個輸入張量 (kTfLiteInt32)，代表輸入 ID。這項輸入簽章需要模型中繼資料中的規則運算式權杖化工具程序單元。
  • 或者一個無形狀的輸入張量 (kTfLiteString)，或形狀 [1] 包含輸入字串。
• 至少一個輸出張量 (kTfLiteFloat32/kTfLiteUint8) 為 [1 x N]，其中 N 是所產生嵌入中的維度數量。