Opsi untuk menjumlahkan masukan di seluruh instans TPU yang direplikasi.
Setiap instance menyediakan masukannya 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]]` akan 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]`.
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
| statis <T memperluas TNumber > CrossReplicaSum <T> | |
| Keluaran <T> | keluaran () Jumlah dari semua input yang didistribusikan. |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
CrossReplicaSum <T> statis publik buat ( Lingkup lingkup, input Operan <T>, Operan < TInt32 > groupAssignment)
Metode pabrik untuk membuat kelas yang membungkus operasi CrossReplicaSum baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| memasukkan | Masukan lokal ke penjumlahan. |
| tugas kelompok | Tensor int32 dengan bentuk [num_groups, num_replicas_per_group]. `group_assignment[i]` mewakili id replika di subgrup ke-i. |
Kembali
- contoh baru CrossReplicaSum