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

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (Object obj)
akhir	hashCode ()
Operasi	op () Mengembalikan mendasari Operation
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama (Object arg0)
Kelas terakhir<?>	getClass ()
ke dalam	hashCode ()
kekosongan akhir	notify ()
kekosongan akhir	notifyAll ()
Rangkaian	toString ()
kekosongan akhir	menunggu (arg0 panjang, int arg1)
kekosongan akhir	menunggu (arg0 panjang)
kekosongan akhir	menunggu ()

Dari antarmuka org.tensorflow.Operand

abstrak Keluaran <T>

asOutput () Mengembalikan pegangan simbolis tensor.

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.