aliran tensor:: operasi:: AkumulatorAmbilGradien
#include <data_flow_ops.h>
Mengekstrak gradien rata-rata dalam ConditionalAccumulator yang diberikan.
Ringkasan
Blok op hingga gradien yang mencukupi (yaitu, lebih dari jumlah_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
- pegangan: Pegangan ke akumulator.
- num_required: Jumlah gradien yang diperlukan sebelum kita mengembalikan agregat.
- dtype: Tipe data akumulasi gradien. Harus sesuai dengan jenis akumulator.
Pengembalian:
-
Output
: Rata-rata akumulasi gradien.
Konstruktor dan Destruktor | |
---|---|
AccumulatorTakeGradient (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_required, DataType dtype) |
Atribut publik | |
---|---|
average | |
operation |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atribut publik
rata-rata
::tensorflow::Output average
operasi
Operation operation
Fungsi publik
AkumulatorAmbilGradien
AccumulatorTakeGradient( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const