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

CollectivePermute

public class akhir CollectivePermute

Op untuk mengubah tensor di seluruh instans TPU yang direplikasi.

Setiap instance memasok inputnya sendiri.

Misalnya, ada 4 instance TPU: `[A, B, C, D]`. Meneruskan source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` mendapatkan output: `[D, A, B, C]`.

Metode Publik

Output <T>
asOutput ()
Mengembalikan pegangan simbolis tensor.
statis <T> CollectivePermute <T>
membuat ( Lingkup lingkup, Operan <T> input, Operan <Integer> sourceTargetPairs)
Metode pabrik untuk membuat kelas yang membungkus operasi CollectivePermute baru.
Output <T>
Output ()
Masukan yang diizinkan.

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 CollectivePermute <T> membuat ( Lingkup lingkup, Operan <T> input, Operan <Integer> sourceTargetPairs)

Metode pabrik untuk membuat kelas yang membungkus operasi CollectivePermute baru.

Parameter
cakupan lingkup saat ini
memasukkan Input lokal yang akan di permutasi. Saat ini hanya mendukung float dan bfloat16.
sumberTargetPairs Tensor dengan bentuk [num_pairs, 2].
Kembali
  • contoh baru CollectivePermute

publik Keluaran <T> Output ()

Masukan yang diizinkan.