tensorflow::ops::SparseApplyMomentum
#include <training_ops.h>'*var' ve '*accum' içindeki ilgili girdileri, ivme şemasına göre güncelleyin.
Özet
Nesterov momentumunu kullanmak istiyorsanız use_nesterov = True olarak ayarlayın.
Yani, notunu aldığımız satırlar için, var ve accum'u aşağıdaki gibi güncelliyoruz:
$$accum = accum * momentum + grad$$
$$var -= lr * accum$$
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- var: Bir Variable() türünden gelmelidir.
- accum: Bir Variable() türünden gelmelidir.
- lr: Öğrenme oranı. Skaler bir değer olmalıdır.
- grad: Eğim.
- İndeksler: Var ve accum'un birinci boyutuna ait indekslerin vektörü.
- 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 | |
|---|---|
SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const SparseApplyMomentum::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::SparseApplyMomentum::Attrs | SparseApplyMomentum için isteğe bağlı öznitelik ayarlayıcıları. |
Genel özellikler
operasyon
Operation operation
dışarı
::tensorflow::Output out
Kamusal işlevler
SeyrekUygulaMomentum
SparseApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum )
SeyrekUygulaMomentum
SparseApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum, const SparseApplyMomentum::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 )