Membagi tensor menjadi `num_split` tensor sepanjang satu dimensi.
Metode Publik
statis <T> Berpisah <T> | |
Iterator < Operan <T >> | iterator () |
Daftar < Keluaran <T >> | Output () Mereka adalah tensor berbentuk identik, yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, di mana ukurannya adalah `nilai.bentuk[split_dim] / num_split`. |
Metode yang Diwarisi
Metode Publik
public static Berpisah <T> membuat ( Lingkup lingkup, Operan <Integer> axis, Operan <T> nilai, Long numSplit)
Metode pabrik untuk membuat kelas yang membungkus operasi Split baru.
Parameter
cakupan | lingkup saat ini |
---|---|
sumbu | 0-D. Dimensi yang akan dipecah. Harus dalam rentang `[-rank(value), rank(value))`. |
nilai | Tensor untuk membelah. |
jumlahSplit | Banyaknya cara membelah. Harus membagi `value.shape[split_dim]` secara merata. |
Kembali
- contoh baru dari Split
public List < Keluaran <T >> keluaran ()
Mereka adalah tensor berbentuk identik, yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, di mana ukurannya adalah `nilai.bentuk[split_dim] / num_split`.