CollectivePermute

kelas akhir publik CollectivePermute

Opsi untuk mengubah posisi tensor di seluruh instance TPU yang direplikasi.

Setiap instance menyediakan masukannya sendiri.

Misalnya, ada 4 instance TPU: `[A, B, C, D]`. Melewati source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` akan menghasilkan keluaran: `[D, A, B, C]`.

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 TType > CollectivePermute <T>
buat ( Lingkup lingkup, masukan Operan <T>, Operan < TInt32 > sourceTargetPairs)
Metode pabrik untuk membuat kelas yang membungkus operasi CollectivePermute baru.
Keluaran <T>
keluaran ()
Masukan yang diijinkan.

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "CollectivePermute"

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.

public static CollectivePermute <T> buat ( Lingkup lingkup, masukan Operan <T>, Operan < TInt32 > sourceTargetPairs)

Metode pabrik untuk membuat kelas yang membungkus operasi CollectivePermute baru.

Parameter
cakupan ruang lingkup saat ini
memasukkan Masukan lokal yang akan diubah. Saat ini hanya mendukung float dan bfloat16.
sourceTargetPairs Tensor dengan bentuk [num_pairs, 2].
Kembali
  • contoh baru dari CollectivePermute

Keluaran publik <T> keluaran ()

Masukan yang diijinkan.