Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

TPUPartitionedOutput

public class akhir TPUPartitionedOutput

Sebuah op yang mendemultipleks tensor untuk di-sharding oleh XLA ke daftar yang dipartisi

output di luar perhitungan XLA.

Kelas Bersarang

kelas TPUPartitionedOutput.Options Atribut opsional untuk TPUPartitionedOutput

Metode Publik

statis <T> TPUPartitionedOutput <T>
membuat ( Lingkup lingkup, Operan <T> masukan, Long numSplits, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.
Iterator < Operan <T >>
Daftar < Keluaran <T >>
Output ()
Daftar input yang dipartisi yang harus memiliki bentuk yang sama.
statis TPUPartitionedOutput.Options
partitionDim (Long partitionDim)

Metode yang Diwarisi

Metode Publik

public static TPUPartitionedOutput <T> membuat ( Lingkup lingkup, Operan <T> masukan, Long numSplits, Options ... pilihan)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.

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

public Iterator < Operan <T >> iterator ()

public List < Keluaran <T >> keluaran ()

Daftar input yang dipartisi yang harus memiliki bentuk yang sama.

public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)

Parameter
partisiDim Sebuah bilangan bulat menggambarkan dimensi mana yang dipartisi.