Mengekstrak gradien rata-rata di ConditionalAccumulator yang diberikan.
Blok op sampai gradien yang cukup (yaitu, lebih dari num_required) telah terakumulasi. Jika akumulator telah mengumpulkan lebih dari num_required gradien, ia mengembalikan rata-rata dari akumulasi gradien. Juga secara otomatis menambah global_step yang direkam di akumulator sebesar 1, dan mengatur ulang agregat ke 0.
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis dari sebuah tensor. |
Keluaran <T> | rata -rata () Rata-rata dari akumulasi gradien. |
statis <T> ResourceAccumulatorTakeGradient <T> |
Metode yang Diwarisi
Metode Publik
Output publik <T> sebagai Output ()
Mengembalikan pegangan simbolis dari sebuah tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static ResourceAccumulatorTakeGradient <T> buat ( Scope scope, Operand <?> handle, Operand <Integer> numRequired, Class<T> dtype)
Metode pabrik untuk membuat kelas yang membungkus operasi ResourceAccumulatorTakeGradient baru.
Parameter
cakupan | lingkup saat ini |
---|---|
menangani | Pegangan ke akumulator. |
jumlahDiperlukan | Jumlah gradien yang diperlukan sebelum kami mengembalikan agregat. |
tipe d | Tipe data akumulasi gradien. Harus sesuai dengan jenis akumulator. |
Kembali
- contoh baru ResourceAccumulatorTakeGradient