aliran tensor:: operasi:: TerapkanTambahkanTanda
#include <training_ops.h>
Perbarui '*var' sesuai dengan pembaruan AddSign.
Ringkasan
m_t <- beta1 * m_{t-1} + (1 - beta1) * g pembaruan <- (alpha + sign_decay * tanda(g) *tanda(m)) * g variabel <- variabel - lr_t * pembaruan
Argumen:
- ruang lingkup: Objek Lingkup
- var: Harus dari Variabel().
- m: Harus dari Variabel().
- lr: Faktor penskalaan. Pasti skalar.
- alpha: Pasti skalar.
- sign_decay: Harus berupa skalar.
- beta: Pasti skalar.
- lulusan: Gradien.
Atribut opsional (lihat Attrs
):
- use_locking: Jika
True
, pembaruan tensor var dan m dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.
Pengembalian:
-
Output
: Sama seperti "var".
Konstruktor dan Destruktor | |
---|---|
ApplyAddSign (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input lr, :: tensorflow::Input alpha, :: tensorflow::Input sign_decay, :: tensorflow::Input beta, :: tensorflow::Input grad) | |
ApplyAddSign (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input lr, :: tensorflow::Input alpha, :: tensorflow::Input sign_decay, :: tensorflow::Input beta, :: tensorflow::Input grad, const ApplyAddSign::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
out |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
UseLocking (bool x) |
Struktur | |
---|---|
tensorflow:: ops:: ApplyAddSign:: Attrs | Penyetel atribut opsional untuk ApplyAddSign . |
Atribut publik
operasi
Operation operation
keluar
::tensorflow::Output out
Fungsi publik
TerapkanTambahkanTanda
ApplyAddSign( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input lr, ::tensorflow::Input alpha, ::tensorflow::Input sign_decay, ::tensorflow::Input beta, ::tensorflow::Input grad )
TerapkanTambahkanTanda
ApplyAddSign( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input lr, ::tensorflow::Input alpha, ::tensorflow::Input sign_decay, ::tensorflow::Input beta, ::tensorflow::Input grad, const ApplyAddSign::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
Gunakan Penguncian
Attrs UseLocking( bool x )