XlaRecvTPUEmbeddingActivations

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

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

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

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

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

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

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

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

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

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

ตัววนซ้ำสาธารณะ < ตัวดำเนินการ <ลอย>> ตัววนซ้ำ ()

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

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