tensör akışı:: işlem:: Momentum Uygula
#include <training_ops.h>
Momentum şemasına göre '*var'ı güncelleyin.
Özet
use_nesterov = True değerini ayarlayın, eğer
Nesterov momentumunu kullanmak istiyorum.
birikim = birikim * momentum + derece var -= lr * birikim
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- var: Bir Variable()'dan olmalıdır.
- accum: Bir Variable()'dan olmalıdır.
- lr: Ölçeklendirme faktörü. Bir skaler olmalı.
- grad: Gradyan.
- momentum: momentum. Bir skaler olmalı.
İsteğe bağlı özellikler (bkz. Attrs
):
- use_locking:
True
ise, var ve accum tensörlerinin güncellenmesi bir kilitle korunacaktır; aksi takdirde davranış tanımsızdır ancak daha az çekişme sergileyebilir. - use_nesterov: Eğer
True
ise, hesaplama derecesine iletilen tensör var - lr * momentum * accum olacaktır, yani sonuçta elde ettiğiniz var aslında var - lr * momentum * accum olacaktır.
İade:
-
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 |
Kamu işlevleri | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Genel statik işlevler | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
Yapılar | |
---|---|
tensorflow:: ops:: ApplyMomentum:: Öznitelikler | ApplyMomentum için isteğe bağlı öznitelik ayarlayıcılar. |
Genel özellikler
operasyon
Operation operation
dışarı
::tensorflow::Output out
Kamu işlevleri
Momentum Uygula
ApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
Momentum Uygula
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
Genel statik işlevler
KullanımKilitleme
Attrs UseLocking( bool x )
KullanımNesterov
Attrs UseNesterov( bool x )