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> | |
Iterator< Operan <T>> | pengulangan () |
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
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. |
---|