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> | |
| 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. |
|---|