CollectiveAllToAllV2

kelas akhir publik CollectiveAllToAllV2

Saling menukar beberapa tensor dengan jenis dan bentuk yang identik.

`is_stateless` berarti setiap operasi tidak memerlukan ketergantungan kontrol ke operasi kolektif lainnya. Dalam hal ini, kunci yang unik pada waktu proses (misalnya `instance_key`) harus digunakan untuk membedakan grup kolektif.

Kelas Bersarang

kelas CollectiveAllToAllV2.Options Atribut opsional untuk CollectiveAllToAllV2

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
CollectiveAllToAllV2.Options statis
Petunjuk komunikasi (Petunjuk komunikasi string)
statis <T memperluas Nomor> CollectiveAllToAllV2 <T>
buat ( Lingkup lingkup, masukan Operand <T>, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi CollectiveAllToAllV2 baru.
Keluaran <T>
data ()
CollectiveAllToAllV2.Options statis
isStateless (Boolean isStateless)
CollectiveAllToAllV2.Options statis
timeoutSeconds (Float timeoutSeconds)

Metode Warisan

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

Public Static CollectiveAllToAllV2.Options CommunicationHint (String CommunicationHint)

public static CollectiveAllToAllV2 <T> buat ( Lingkup cakupan , input Operand <T>, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi CollectiveAllToAllV2 baru.

Parameter
cakupan ruang lingkup saat ini
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari CollectiveAllToAllV2

publik Keluaran <T> data ()

public static CollectiveAllToAllV2.Options isStateless (Boolean isStateless)

public static CollectiveAllToAllV2.Options timeoutSeconds (Float timeoutSeconds)