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 >> | iterator () |
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 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. |
---|