XlaSparseDenseMatmulGradWithAdamAndCsrInput

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

Các lớp lồng nhau

lớp học XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options Thuộc tính tùy chọn cho XlaSparseDenseMatmulGradWithAdamAndCsrInput

Phương pháp công cộng

tĩnh XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options
clipWeightMax (Float clipWeightMax)
tĩnh XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options
clipWeightMin (clip nổiWeightMin)
tĩnh XlaSparseDenseMatmulGradWithAdamAndCsrInput
tạo ( Phạm vi phạm vi , Toán hạng <Integer> rowPointers, Toán hạng <Integer> đã sắp xếpSampleIds, Toán hạng <Integer> đã sắp xếpTokenIds, Toán hạng <Float> đã sắp xếpGains, Toán hạng <Float> activateGradients, Toán hạng <Float> learningRate, Toán hạng <Float> embeddingTable, Toán hạng <Float > khoảnh khắc, Vận tốc toán hạng <Float>, Toán hạng <Integer> numMinibatchesPerPhysicalSparseCore, Boolean useSumInsideSqrt, Float beta1, Float beta2, Float epsilon, String tableName, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao gồm thao tác XlaSparseDenseMatmulGradWithAdamAndCsrInput mới.
Đầu ra <Float>
Đầu ra <Float>
Đầu ra <Float>

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

Phương pháp công cộng

tĩnh công khai XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options clipWeightMax (Float clipWeightMax)

tĩnh công khai XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options clipWeightMin (Float clipWeightMin)

public static XlaSparseDenseMatmulGradWithAdamAndCsrInput create ( Phạm vi phạm vi , Toán hạng <Integer> rowPointers, Toán hạng <Integer> đã sắp xếpSampleIds, Toán hạng <Integer> đã sắp xếpTokenIds, Toán hạng <Float> đã sắp xếpGains, Toán hạng <Float> kích hoạtGradients, Toán hạng <Float> learningRate, Toán hạng <Float> embeddingTable, Toán hạng <Float> khoảnh khắc, Vận tốc toán hạng <Float>, Toán hạng <Integer> numMinibatchesPerPhysicalSparseCore, Boolean useSumInsideSqrt, Float beta1, Float beta2, Float epsilon, Tên bảng chuỗi, Tùy chọn... tùy chọn)

Phương thức xuất xưởng để tạo một lớp bao gồm thao tác XlaSparseDenseMatmulGradWithAdamAndCsrInput mới.

Thông số
phạm vi phạm vi hiện tại
tùy chọn mang các giá trị thuộc tính tùy chọn
Trả lại
  • một phiên bản mới của XlaSparseDenseMatmulGradWithAdamAndCsrInput

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

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

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