Sebuah operasi yang mengelompokkan daftar input yang dipartisi menjadi satu. Mendukung sharding ND.
Kelas Bersarang
kelas | TPUPartitionedInputV2.Options | Atribut opsional untuk TPUPartitionedInputV2 |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
statis <T> TPUPartitionedInputV2 <T> | |
TPUPartitionedInputV2.Options statis | isPacked (Boolean isPacked) |
Keluaran <T> | keluaran () Pegangan yang mewakili bentuk penuh tensor yang dipartisi. |
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 TPUPartitionedInputV2 <T> buat ( Lingkup cakupan , input Iterable< Operand <T>>, Daftar<Panjang> partisiDims, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedInputV2 baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
masukan | Daftar input yang dipartisi yang harus memiliki bentuk yang sama. |
partisiDim | Daftar bilangan bulat yang menjelaskan bagaimana setiap dimensi dipartisi. Kekosongan menunjukkan masukan direplikasi. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru TPUPartitionedInputV2
TPUPartitionedInputV2.Options statis publik isPacked (Boolean isPacked)
Parameter
sudah dikemas | Menunjukkan apakah input merupakan sumber daya yang dikemas. |
---|