tensor 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:
Trueise, 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
Trueise, 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 )