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> | |
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