tensorflow :: ops :: ResourceSparseApplyKerasMomentum
#include <training_ops.h>
Perbarui entri yang relevan di '* var' dan '* akumulasi' sesuai dengan skema momentum.
Ringkasan
Setel use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.
Itu untuk baris yang telah kami tingkatkan, kami memperbarui var dan akumulasi sebagai berikut:
akum = akum * momentum - lr * grad var + = akum
Argumen:
- scope: Objek Scope
- var: Harus dari Variabel ().
- akumulasi: Harus dari Variabel ().
- lr: Kecepatan pembelajaran. Harus skalar.
- grad: Gradien.
- indeks: Vektor indeks ke dalam dimensi pertama var dan akum.
- 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
Pembuat dan Penghancur | |
---|---|
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const ResourceSparseApplyKerasMomentum::Attrs & attrs) |
Atribut publik | |
---|---|
operation |
Fungsi publik | |
---|---|
operator::tensorflow::Operation () const |
Fungsi statis publik | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
Structs | |
---|---|
tensorflow :: ops :: ResourceSparseApplyKerasMomentum :: Attrs | Penyetel atribut opsional untuk ResourceSparseApplyKerasMomentum . |
Atribut publik
operasi
Operation operation
Fungsi publik
ResourceSparseApplyKerasMomentum
ResourceSparseApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum )
ResourceSparseApplyKerasMomentum
ResourceSparseApplyKerasMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum, const ResourceSparseApplyKerasMomentum::Attrs & attrs )
operator :: tensorflow :: Operasi
operator::tensorflow::Operation() const
Fungsi statis publik
UseLocking
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )