aliran tensor:: operasi:: Akumulator JarangAmbil Gradien
#include <data_flow_ops.h>
Mengekstrak gradien renggang rata-rata dalam SparseConditionalAccumulator .
Ringkasan
Operasi akan memblokir hingga gradien yang cukup (yaitu, lebih dari jumlah_yang diperlukan) telah terakumulasi. Jika akumulator telah mengumpulkan lebih dari num_required gradien, akumulator akan mengembalikan rata-rata akumulasi gradien. Juga secara otomatis menambah global_step yang tercatat di akumulator sebesar 1, dan menyetel ulang agregat ke 0.
Argumen:
- ruang lingkup: Objek Lingkup
- handle: Pegangan ke SparseConditionalAccumulator .
- num_required: Jumlah gradien yang diperlukan sebelum kita mengembalikan agregat.
- dtype: Tipe data akumulasi gradien. Harus sesuai dengan jenis akumulator.
Pengembalian:
- Indeks
Output
: Indeks rata-rata akumulasi gradien renggang. - Nilai
Output
: Nilai rata-rata akumulasi gradien renggang. - Bentuk
Output
: Bentuk rata-rata akumulasi gradien renggang.
Konstruktor dan Destruktor | |
---|---|
SparseAccumulatorTakeGradient (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_required, DataType dtype) |
Atribut publik | |
---|---|
indices | |
operation | |
shape | |
values |
Atribut publik
indeks
::tensorflow::Output indices
operasi
Operation operation
membentuk
::tensorflow::Output shape
nilai-nilai
::tensorflow::Output values
Fungsi publik
Akumulator JarangAmbil Gradien
SparseAccumulatorTakeGradient( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype )