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]`.
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
statis <T> Permute Kolektif <T> | |
Keluaran <T> | keluaran () Masukan yang diijinkan. |
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 CollectivePermute <T> buat ( Lingkup lingkup, input Operand <T>, Operand <Integer> 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