SendTPUEmbeddingGradients

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

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

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

SendTPUEmbeddingGradients แบบคงที่
สร้าง (ขอบเขต ขอบเขต , Iterable< Operand <Float>> อินพุต, Iterable< Operand <Float>> LearningRates, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SendTPUEmbeddingGradients ใหม่

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

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

สร้าง SendTPUEmbeddingGradients แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable < Operand <Float>> อินพุต, Iterable < Operand <Float>> LearningRates, การกำหนดค่าสตริง)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
อินพุต TensorList ของการไล่ระดับสีสำหรับอัปเดตตารางที่ฝัง อาร์กิวเมนต์นี้มีความยาวและรูปร่างเท่ากันกับค่าที่ส่งคืนของ RecvTPUEmbeddingActivations แต่มีการไล่ระดับสีของการสูญเสียของโมเดลที่เกี่ยวข้องกับการเปิดใช้งานการฝัง ตารางการฝังได้รับการอัปเดตจากการไล่ระดับสีเหล่านี้ผ่านเครื่องมือเพิ่มประสิทธิภาพที่ระบุในการกำหนดค่าการฝัง TPU ที่มอบให้กับ tpu.initialize_system
อัตราการเรียนรู้ TensorList ของสเกลาร์ float32 หนึ่งรายการสำหรับแท็กอัตราการเรียนรู้แบบไดนามิกแต่ละรายการ: ดูความคิดเห็นใน //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto หลายตารางสามารถแชร์แท็กอัตราการเรียนรู้แบบไดนามิกเดียวกันตามที่ระบุไว้ในการกำหนดค่า หากอัตราการเรียนรู้ของตารางทั้งหมดคงที่ รายการนี้ควรว่างเปล่า
กำหนดค่า โปรโตคอล TPUEmbeddingConfiguration แบบอนุกรม
การส่งคืน
  • อินสแตนซ์ใหม่ของ SendTPUEmbeddingGradients