tensorflow :: ops :: ResourceApplyKerasMomentum
#include <training_ops.h>
Perbarui '* var' sesuai dengan skema momentum.
Ringkasan
Setel use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.
akum = akum * momentum - lr * grad var + = akum
Argumen:
- scope: Objek Scope
- var: Harus dari Variabel ().
- akumulasi: Harus dari Variabel ().
- lr: Faktor penskalaan. Harus skalar.
- grad: Gradien.
- momentum: Momentum. Harus skalar.
Atribut opsional (lihat Attrs
):
- use_locking: Jika
True
, pembaruan tensor var dan akumulator akan dilindungi oleh kunci; jika tidak, perilakunya tidak ditentukan, tetapi mungkin menunjukkan lebih sedikit pertentangan. - use_nesterov: Jika
True
, tensor yang dilewatkan untuk menghitung grad akan menjadi var + momentum * akumulasi, jadi pada akhirnya, var yang Anda dapatkan sebenarnya adalah var + momentum * akumulasi.
Pengembalian:
-
Operation
dibuat
Pembangun dan Penghancur | |
---|---|
ResourceApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum) | |
ResourceApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum, const ResourceApplyKerasMomentum::Attrs & attrs) |
Atribut publik | |
---|---|
operation |
Fungsi publik | |
---|---|
operator::tensorflow::Operation () const |
Fungsi statis publik | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
Structs | |
---|---|
tensorflow :: ops :: ResourceApplyKerasMomentum :: Attrs | Penyetel atribut opsional untuk ResourceApplyKerasMomentum . |
Atribut publik
operasi
Operation operation
Fungsi publik
ResourceApplyKerasMomentum
ResourceApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
ResourceApplyKerasMomentum
ResourceApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ResourceApplyKerasMomentum::Attrs & attrs )
operator :: tensorflow :: Operasi
operator::tensorflow::Operation() const
Fungsi statis publik
UseLocking
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )