TextEmbedder

ดำเนินการดึงข้อมูลการฝังในข้อความ

API นี้ควรเป็นโมเดล TFLite ที่มีข้อมูลเมตาของโมเดล TFLite (ไม่บังคับ)

ต้องมีข้อมูลเมตาสำหรับโมเดลที่มี Tensor อินพุต int32 เนื่องจากมีหน่วยกระบวนการอินพุตสำหรับ Tokenizer ของโมเดล ไม่จำเป็นต้องมีข้อมูลเมตาสำหรับโมเดลที่มี Tensor อินพุตสตริง

• Tensor อินพุต
  • Tensor อินพุต 3 รายการ (kTfLiteInt32) ของรูปร่าง [batch_size x bert_max_seq_len] ที่แสดงถึงรหัสอินพุต รหัสมาสก์ และรหัสกลุ่ม ลายเซ็นของอินพุตนี้ต้องมีหน่วยประมวลผล Bert Tokenizer ในข้อมูลเมตาของโมเดล
  • หรือ Tensor อินพุต (kTfLiteInt32) ของรูปร่าง [batch_size x max_seq_len] ที่แทนรหัสอินพุต ลายเซ็นอินพุตนี้ต้องมีหน่วยประมวลผล Regex Tokenizer ในข้อมูลเมตาของโมเดล
  • หรือ Tensor อินพุต (kTfLiteString) ที่ไม่มีรูปร่างหรือมีรูปร่าง [1] ที่มีสตริงอินพุต
• Tensor เอาต์พุตอย่างน้อย 1 รายการ (kTfLiteFloat32/kTfLiteUint8) ที่มีรูปร่าง [1 x N] โดยที่ N คือจำนวนมิติข้อมูลในการฝังที่สร้างขึ้น