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