XlaSparseCoreAdam

lớp cuối cùng công khai XlaSparseCoreAdam

Phương pháp công khai

XlaSparseCoreAdam tĩnh
tạo ( Phạm vi phạm vi, Bảng nhúng toán hạng <Float>, Chỉ số toán hạng <Số nguyên>, Độ dốc toán hạng <Float>, Tốc độ học toán hạng <Float>, Động lượng toán hạng <Float>, Vận tốc toán hạng <Float>, Toán hạng <Float> beta1, Toán hạng <Float> > beta2, Toán hạng <Float> epsilon, Tính năng dàiWidth, Boolean useSumInsideSqrt)
Phương thức xuất xưởng để tạo một lớp bao bọc hoạt động XlaSparseCoreAdam mới.
Đầu ra <Float>
Đầu ra <Float>
Đầu ra <Float>

Phương pháp kế thừa

Phương pháp công khai

public static XlaSparseCoreAdam tạo ( Phạm vi phạm vi, Bảng nhúng toán hạng <Float>, chỉ số Toán hạng <Số nguyên>, Độ dốc toán hạng <Float>, Tốc độ học toán toán hạng, Động lượng toán hạng <Float>, Vận tốc toán hạng <Float>, Toán hạng <Float> beta1, Toán hạng <Float> beta2, Toán hạng <Float> epsilon, Tính năng dàiWidth, Boolean useSumInsideSqrt)

Phương thức xuất xưởng để tạo một lớp bao bọc hoạt động XlaSparseCoreAdam mới.

Thông số
phạm vi phạm vi hiện tại
Trả lại
  • một phiên bản mới của XlaSparseCoreAdam

Đầu ra công khai <Float> đã cập nhậtEmbeddingTable ()

Đầu ra công khai <Float> được cập nhậtMomentum ()

Đầu ra công khai <Float> đã cập nhậtVelocity ()