Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

CrossReplicaSum

public class akhir CrossReplicaSum

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>
membuat ( Lingkup lingkup, Operan <T> input, Operan <Integer> groupAssignment)
Metode pabrik untuk membuat kelas yang membungkus operasi CrossReplicaSum baru.
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

publik Keluaran <T> Output ()

Jumlah dari semua input yang didistribusikan.