คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

XlaRecvTPUEmbeddingActivations

คลาสสุดท้ายสาธารณะ XlaRecvTPUEmbeddingActivations

op ที่ได้รับการเปิดใช้งานการฝังบน TPU

ระบบ TPU ดำเนินการค้นหาและการรวมการฝัง ผลลัพธ์ของการรวมเหล่านี้สามารถมองเห็นได้ในกราฟ Tensorflow เป็นผลลัพธ์ของ XlaRecvTPUEmbeddingActivations Op op นี้ส่งคืนรายการที่มีหนึ่ง Tensor ของการเปิดใช้งานต่อตารางที่ระบุในโมเดล

วิธีการสาธารณะ

คงที่ XlaRecvTPUEmbeddingActivations
สร้าง ( ขอบเขต ขอบเขต, ตัวถูกดำเนินการ <?> deduplicationData, numTables แบบยาว, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaRecvTPUEmbeddingActivations ใหม่
Iterator < ตัวถูกดำเนินการ <Float>>
รายการ< เอาท์พุต <โฟลต>>
เอาท์พุท ()
TensorList ของการเปิดใช้งานการฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล

วิธีการสืบทอด

วิธีการสาธารณะ

XlaRecvTPUEmbeddingActivations สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต, ตัวถูกดำเนินการ <?> deduplicationData, Long numTables , String config)

วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaRecvTPUEmbeddingActivations ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
deduplicationData Tensor with type=DT_VARIANT ที่มีข้อมูลการขจัดความซ้ำซ้อน เทนเซอร์คือทูเพิลที่ซ้อนกัน XLA ที่มีองค์ประกอบ N (โดยที่ N คืออัตราส่วนของจำนวนการฝังต่อเทนเซอร์คอร์ต่อชิป TPU) แต่ละองค์ประกอบของทูเพิลที่ซ้อนกันคือทูเพิลของเทนเซอร์อันดับ 1 เมตริกซ์แต่ละตัวมีดัชนี (DT_UIT32) สำหรับการฝังการค้นหาบน TensorCore หรือน้ำหนัก (DT_FLOAT) เพื่อนำไปใช้กับเอาต์พุตของการดำเนินการค้นหาแบบฝัง
numTables จำนวนเทนเซอร์การเปิดใช้งานเอาต์พุต หากมีตัวอธิบายคุณลักษณะในการกำหนดค่าการฝัง tpu จะเท่ากับจำนวนคุณลักษณะ มิฉะนั้นจะเท่ากับจำนวนตารางการฝังในโมเดล
config โปรโต TPUEmbeddingConfiguration แบบอนุกรม
คืนสินค้า
  • อินสแตนซ์ใหม่ของ XlaRecvTPUEmbeddingActivations

ตัววนซ้ำ สาธารณะ< Operand <Float>> iterator ()

รายการสาธารณะ < เอาท์พุต <โฟลต>> เอาท์พุท ()

TensorList ของการเปิดใช้งานการฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล