Op untuk menjumlahkan input di seluruh instance TPU yang direplikasi.
Setiap instance memasok inputnya sendiri.
Misalnya, ada 8 instance TPU: `[A, B, C, D, E, F, G, H]`. Melewati group_assignment=`[[0,2,4,6],[1,3,5,7]]` menetapkan `A, C, E, G` sebagai grup 0, dan `B, D, F, H` sebagai grup 1. Jadi kita mendapatkan output: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`.
Metode Publik
Output <T> | asOutput () Mengembalikan pegangan simbolis tensor. |
statis <T meluas Nomor> CrossReplicaSum <T> | |
Output <T> | Output () Jumlah dari semua input yang didistribusikan. |
Metode yang Diwarisi
Metode Publik
publik Keluaran <T> asOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static CrossReplicaSum <T> membuat ( Lingkup lingkup, Operan <T> input, Operan <Integer> groupAssignment)
Metode pabrik untuk membuat kelas yang membungkus operasi CrossReplicaSum baru.
Parameter
cakupan | lingkup saat ini |
---|---|
memasukkan | Input lokal ke jumlah. |
tugas grup | Tensor int32 dengan bentuk [num_groups, num_replicas_per_group]. `group_assignment[i]` mewakili id replika di subgrup ke-i. |
Kembali
- contoh baru CrossReplicaSum