ResourceApplyMomentum kelas akhir publik
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 | ResourceApplyMomentum.Options | Atribut opsional untuk ResourceApplyMomentum | |
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| statis <T memperluas TType > ResourceApplyMomentum | |
| ResourceApplyMomentum.Options statis | useLocking (penguncian penggunaan Boolean) |
| ResourceApplyMomentum.Options statis | useNesterov (Boolean useNesterov) |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "ResourceApplyMomentum"
Metode Publik
pembuatan ResourceApplyMomentum statis publik ( Lingkup cakupan , Operan <?> var, Operan <?> accum, Operan <T> lr, Operan <T> grad, Operan <T> momentum, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi ResourceApplyMomentum 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 ResourceApplyMomentum
ResourceApplyMomentum.Options statis publik useLocking (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. |
|---|
ResourceApplyMomentum.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. |
|---|