ApplyMomentum

kelas akhir publik ApplyMomentum

Perbarui '*var' sesuai dengan skema momentum.

Tetapkan use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.

accum = accum * momentum + lulusan var -= lr * accum

Kelas Bersarang

kelas TerapkanMomentum.Opsi Atribut opsional untuk ApplyMomentum

Konstanta

Rangkaian OP_NAME Nama operasi ini dikenal dengan mesin inti TensorFlow

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
statis <T memperluas TType > ApplyMomentum <T>
buat ( Lingkup lingkup, Operand <T> var, Operand <T> accum, Operand <T> lr, Operand <T> grad, Operand <T> momentum, Pilihan... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi ApplyMomentum baru.
Keluaran <T>
keluar ()
Sama seperti "var".
ApplyMomentum.Options statis
useLocking (penguncian penggunaan Boolean)
ApplyMomentum.Options statis
useNesterov (Boolean useNesterov)

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "ApplyMomentum"

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolis tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static ApplyMomentum <T> buat ( Lingkup lingkup, Operand <T> var, Operand <T> accum, Operand <T> lr, Operand <T> grad, Operand <T> momentum, Opsi... pilihan)

Metode pabrik untuk membuat kelas yang membungkus operasi ApplyMomentum baru.

Parameter
cakupan ruang lingkup saat ini
var Harus dari Variabel().
akumulasi Harus dari Variabel().
lr Faktor skala. Pasti skalar.
lulusan Gradien.
momentum momentum. Pasti skalar.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari ApplyMomentum

Keluaran publik <T> keluar ()

Sama seperti "var".

ApplyMomentum.Options useLocking statis publik (Boolean useLocking)

Parameter
gunakan Penguncian Jika `True`, pembaruan tensor var dan accum akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.

ApplyMomentum.Options statis publik useNesterov (Boolean useNesterov)

Parameter
gunakanNesterov Jika `True`, tensor yang diteruskan ke komputasi grad adalah var - lr * momentum * accum, jadi pada akhirnya, var yang Anda dapatkan sebenarnya adalah var - lr * momentum * accum.