TPUPartitionedOutput

TPUPartitionedOutput kelas akhir publik

Sebuah operasi yang mendemultiplexing tensor untuk dipecah oleh XLA ke daftar yang dipartisi

keluaran di luar komputasi XLA.

Kelas Bersarang

kelas TPUPartitionedOutput.Options Atribut opsional untuk TPUPartitionedOutput

Metode Publik

statis <T> TPUPartitionedOutput <T>
buat ( Lingkup lingkup, input Operan <T>, numSplits panjang, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.
Iterator< Operan <T>>
Daftar< Keluaran <T>>
keluaran ()
Daftar input yang dipartisi yang harus memiliki bentuk yang sama.
TPUPartitionedOutput.Options statis
partisiDim (Partisi panjangDim)

Metode Warisan

Metode Publik

public static TPUPartitionedOutput <T> buat ( Lingkup lingkup, input Operan <T>, Long numSplits, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.

Parameter
cakupan ruang lingkup saat ini
masukan Tensor yang mewakili bentuk penuh tensor yang dipartisi.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari TPUPartitionedOutput

Iterator publik< Operan <T>> iterator ()

Daftar publik< Keluaran <T>> keluaran ()

Daftar input yang dipartisi yang harus memiliki bentuk yang sama.

publik statis TPUPartitionedOutput.Options partisiDim (Partisi panjangDim)

Parameter
partisiDim Bilangan bulat menjelaskan dimensi mana yang dipartisi.