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> |
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.
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