XlaSendTPUEmbeddingGradients

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

op ที่ทำการอัปเดตการไล่ระดับสีของตารางการฝัง

อาร์กิวเมนต์การไล่ระดับสีคือ TensorList ที่มีความยาวและรูปร่างเหมือนกันกับค่าส่งคืนของ XlaRecvTPUEmbeddingActivations แต่มีการไล่ระดับสีของการสูญเสียของโมเดลที่เกี่ยวข้องกับการเปิดใช้งานการฝัง ตารางการฝังได้รับการอัปเดตจากการไล่ระดับสีเหล่านี้ผ่านเครื่องมือเพิ่มประสิทธิภาพที่ระบุในโปรโต TPUEmbeddingConfiguration ที่มอบให้กับ tpu.initialize_system

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

คงที่ XlaSendTPUEmbeddingGradients

สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <Float>> gradients, Iterable< Operand <Float>> learningRates, Operand <?> deduplicationData, String config) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaSendTPUEmbeddingGradients ใหม่

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

จากคลาส org.tensorflow.op.PrimitiveOp

บูลีนสุดท้าย	เท่ากับ (วัตถุ obj)
intสุดท้าย	แฮชโค้ด ()
การดำเนินการ	ออฟ () ส่งกลับ Operation พื้นฐาน
สตริงสุดท้าย	toString ()

จากคลาส java.lang.Object

บูลีน	เท่ากับ (วัตถุ arg0)
คลาสสุดท้าย<?>	getClass ()
int	แฮชโค้ด ()
สุดท้ายเป็นโมฆะ	แจ้ง ()
สุดท้ายเป็นโมฆะ	แจ้งทั้งหมด ()
สตริง	toString ()
สุดท้ายเป็นโมฆะ	รอ (arg ยาว, int arg1)
สุดท้ายเป็นโมฆะ	รอ (ยาว arg0)
สุดท้ายเป็นโมฆะ	รอ ()

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

XlaSendTPUEmbeddingGradients สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <Float>> gradients, Iterable< Operand <Float>> learningRates, Operand <?> deduplicationData, String config)

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

พารามิเตอร์

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

คืนสินค้า

• อินสแตนซ์ใหม่ของ XlaSendTPUEmbeddingGradients