XlaSparseCoreAdam

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

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

XlaSparseCoreAdam แบบคงที่
สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <Float> embeddingTable, ตัวดำเนินการ <Integer> ดัชนี, ตัวดำเนินการ <Float> การไล่ระดับสี, ตัวดำเนินการ <Float> อัตราการเรียนรู้, ตัวดำเนินการ <Float> โมเมนตัม, ตัวดำเนินการ <Float> ความเร็ว, ตัว ถูกดำเนินการ <Float> beta1, ตัวถูกดำเนินการ <Float > beta2, ตัวดำเนินการ <Float> เอปไซลอน, คุณลักษณะกว้างยาว, บูลีน useSumInsideSqrt)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaSparseCoreAdam ใหม่
เอาท์พุต <โฟลต>
เอาท์พุต <โฟลต>
เอาท์พุต <โฟลต>

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

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

การสร้าง XlaSparseCoreAdam แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , ตัวดำเนินการ <Float> embeddingTable, ตัวดำเนินการ <Integer> ดัชนี, ตัวดำเนิน การ <Float> การไล่ระดับสี, ตัวดำเนินการ <Float> อัตราการเรียนรู้, ตัวดำเนินการ <Float> โมเมนตัม, ตัว ดำเนินการ <Float> ความเร็ว, ตัวดำเนินการ <Float> beta1, ตัวดำเนินการ <Float> beta2, ตัวดำเนินการ <Float> เอปไซลอน, คุณลักษณะกว้างยาว, บูลีน useSumInsideSqrt)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ XlaSparseCoreAdam

เอาท์พุท สาธารณะ <Float> อัปเดต EmbeddingTable ()

เอาท์พุท สาธารณะ <ลอย> ปรับปรุงโมเมนตัม ()

เอาท์พุท สาธารณะ <Float> อัปเดตความเร็ว ()