tensorflow::ops::ResourceSparseApplyKerasMomentum
#include <training_ops.h>Perbarui entri yang relevan di '*var' dan '*accum' sesuai dengan skema momentum.
Ringkasan
Atur use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.
Artinya, untuk baris yang sudah memiliki nilai kelulusan, kita memperbarui variabel dan akumulasi sebagai berikut:
akumulasi = akumulasi * momentum - lr * lulusan var += akumulasi
Argumen:
- Cakupan: Objek Cakupan
- var: Seharusnya berasal dari Variable().
- accum: Seharusnya berasal dari Variable().
- lr: Tingkat pembelajaran. Harus berupa bilangan skalar.
- grad: Gradien.
- indeks: Sebuah vektor indeks ke dimensi pertama dari var dan accum.
- momentum: Momentum. Harus berupa besaran skalar.
Atribut opsional (lihat Attrs ):
- use_locking: Jika
True, pembaruan tensor var dan accum akan dilindungi oleh kunci; jika tidak, perilakunya tidak terdefinisi, tetapi mungkin menunjukkan lebih sedikit persaingan. - use_nesterov: Jika
True, tensor yang diteruskan ke compute grad akan menjadi var + momentum * accum, jadi pada akhirnya, var yang Anda dapatkan sebenarnya adalah var + momentum * accum.
Mengembalikan:
-
Operationyang dibuat
Konstruktor dan Destruktor | |
|---|---|
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const ResourceSparseApplyKerasMomentum::Attrs & attrs) |
Atribut publik | |
|---|---|
operation | |
Fungsi publik | |
|---|---|
operator::tensorflow::Operation () const | |
Fungsi statis publik | |
|---|---|
UseLocking (bool x) | |
UseNesterov (bool x) | |
Struktur | |
|---|---|
| tensorflow::ops::ResourceSparseApplyKerasMomentum::Attrs | Pengaturan atribut opsional untuk ResourceSparseApplyKerasMomentum . |
Atribut publik
operasi
Operation operation
Fungsi publik
Sumber DayaJalankanKerasMomentum
ResourceSparseApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum )
Sumber DayaJalankanKerasMomentum
ResourceSparseApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum, const ResourceSparseApplyKerasMomentum::Attrs & attrs )
operator::tensorflow::Operasi
operator::tensorflow::Operation() const
Fungsi statis publik
GunakanPenguncian
Attrs UseLocking( bool x )
Gunakan Nesterov
Attrs UseNesterov( bool x )