ResourceAccumulatorTakeGradient

ResourceAccumulatorTakeGradient kelas akhir publik

Mengekstrak gradien rata-rata di Akumulator Bersyarat yang diberikan.

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.

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Keluaran <T>
rata-rata ()
Rata-rata akumulasi gradien.
statis <T> ResourceAccumulatorTakeGradient <T>
buat ( Lingkup lingkup, Operan <?> pegangan, Operan <Integer> numRequired, Kelas<T> dtype)
Metode pabrik untuk membuat kelas yang membungkus operasi ResourceAccumulatorTakeGradient baru.

Metode Warisan

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

Output publik <T> rata-rata ()

Rata-rata akumulasi gradien.

public static ResourceAccumulatorTakeGradient <T> buat ( Lingkup lingkup, Operan <?> pegangan, Operan <Integer> numRequired, Kelas<T> dtype)

Metode pabrik untuk membuat kelas yang membungkus operasi ResourceAccumulatorTakeGradient baru.

Parameter
cakupan ruang lingkup saat ini
menangani Pegangan ke akumulator.
jumlah yang diperlukan Jumlah gradien yang diperlukan sebelum kita mengembalikan agregat.
tipe Tipe data akumulasi gradien. Harus sesuai dengan jenis akumulator.
Kembali
  • contoh baru ResourceAccumulatorTakeGradient