XlaRecvTPUEmbeddingDeduplicationData

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

รับข้อมูลการขจัดความซ้ำซ้อน (ดัชนีและน้ำหนัก) จากคอร์ที่ฝัง

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

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

เอาต์พุต <วัตถุ>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
XlaRecvTPUEmbeddingDeduplicationData แบบคงที่
สร้าง (ขอบเขต ขอบเขต , การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaRecvTPUEmbeddingDeduplicationData ใหม่
เอาท์พุต <?>

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

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

เอาท์ พุท สาธารณะ <Object> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สร้าง XlaRecvTPUEmbeddingDeduplicationData สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , การกำหนดค่าสตริง)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
กำหนดค่า โปรโตคอล TPUEmbeddingConfiguration แบบอนุกรม
การส่งคืน
  • อินสแตนซ์ใหม่ของ XlaRecvTPUEmbeddingDeduplicationData

เอาท์พุท สาธารณะ <?> เอาท์พุท ()