tensorflow::ops::ApplyMomentum
#include <training_ops.h>'*var' değişkenini momentum şemasına göre güncelleyin.
Özet
Eğer use_nesterov = True ise, bunu ayarlayın.
Nesterov momentumunu kullanmak istiyorum.
birikim = birikim * momentum + derece var -= lr * birikim
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- var: Bir Variable() türünden gelmelidir.
- accum: Bir Variable() türünden gelmelidir.
- lr: Ölçeklendirme faktörü. Skaler bir değer olmalıdır.
- grad: Eğim.
- momentum: Momentum. Skaler bir sayı olmalıdır.
İsteğe bağlı özellikler (bkz. Attrs ):
- use_locking:
Trueise, var ve accum tensörlerinin güncellenmesi bir kilit ile korunacaktır; aksi takdirde davranış tanımsızdır, ancak daha az çekişme gösterebilir. - use_nesterov: Eğer
Trueise, grad hesaplamasına geçirilen tensör var - lr * momentum * accum olacaktır, dolayısıyla sonuçta elde ettiğiniz var aslında var - lr * momentum * accum olacaktır.
İadeler:
-
Output: "var" ile aynı.
Yapıcılar ve Yıkıcılar | |
|---|---|
ApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum) | |
ApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum, const ApplyMomentum::Attrs & attrs) |
Genel özellikler | |
|---|---|
operation | |
out | |
Kamusal işlevler | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Herkese açık statik fonksiyonlar | |
|---|---|
UseLocking (bool x) | |
UseNesterov (bool x) | |
Yapılar | |
|---|---|
| tensorflow::ops::ApplyMomentum::Attrs | ApplyMomentum için isteğe bağlı özellik ayarlayıcıları. |
Genel özellikler
operasyon
Operation operation
dışarı
::tensorflow::Output out
Kamusal işlevler
ApplyMomentum
ApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
ApplyMomentum
ApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ApplyMomentum::Attrs & attrs )
düğüm
::tensorflow::Node * node() const
operatör::tensorflow::Giriş
operator::tensorflow::Input() const
operatör::tensorflow::Çıktı
operator::tensorflow::Output() const
Herkese açık statik fonksiyonlar
KullanımKilitleme
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )