XlaSparseCoreAdam

kelas akhir publik XlaSparseCoreAdam

Metode Publik

XlaSparseCoreAdam statis
buat ( Scope scope, Operand <Float> embeddingTable, indeks Operand <Integer>, gradien Operand <Float>, LearningRate Operand <Float>, momentum Operand <Float>, kecepatan Operand <Float>, Operand <Float> beta1, Operand <Float > beta2, Operan <Float> epsilon, FeatureWidth panjang, Boolean useSumInsideSqrt)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSparseCoreAdam baru.
Keluaran <Mengambang>
Keluaran <Mengambang>
Keluaran <Mengambang>

Metode Warisan

Metode Publik

public static XlaSparseCoreAdam membuat ( Lingkup cakupan, Operand <Float> embeddingTable, indeks Operand <Integer>, gradien Operand <Float>, Operand <Float> learningRate, momentum Operand <Float>, kecepatan Operand <Float>, Operand <Float> beta1, Operan <Float> beta2, Operan <Float> epsilon, Long featureWidth, Boolean useSumInsideSqrt)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaSparseCoreAdam baru.

Parameter
cakupan ruang lingkup saat ini
Kembali
  • contoh baru dari XlaSparseCoreAdam

Output publik <Float> diperbaruiEmbeddingTable ()

Output publik <Float> updateMomentum ()

Output publik <Float> diperbaruiVelocity ()