TPUPartitionedInputV2

TPUPartitionedInputV2 kelas akhir publik

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>
buat ( Lingkup lingkup, input Iterable< Operand <T>>, Daftar<Panjang> partisiDims, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedInputV2 baru.
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.

Keluaran publik <T> keluaran ()

Pegangan yang mewakili bentuk penuh tensor yang dipartisi.