kelas akhir publik XlaSpmdShardToFullShape
Sebuah operasi yang digunakan oleh partisi XLA SPMD untuk beralih dari partisi manual ke
partisi otomatis. Ini mengubah input berbentuk shard yang dipartisi secara manual menjadi tensor berbentuk penuh untuk dipartisi secara otomatis dengan sharding yang sama dengan yang digunakan oleh partisi manual.
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 > XlaSpmdShardToFullShape <T> | |
| Keluaran <T> | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "XlaSpmdShardToFullShape"
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 XlaSpmdShardToFullShape <T> buat ( Lingkup lingkup , masukan Operan <T>, String manualSharding, Bentuk fullShape)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSpmdShardToFullShape baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|
Kembali
- contoh baru dari XlaSpmdShardToFullShape