SplitV kelas akhir publik
Membagi tensor menjadi tensor `num_split` sepanjang satu dimensi.
Metode Publik
statis <T, U memperluas Angka> SplitV <T> | |
Iterator< Operan <T>> | pengulangan () |
Daftar< Keluaran <T>> | keluaran () Tensor yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, yang ukurannya adalah `size_splits[i]`. |
Metode Warisan
Metode Publik
public static SplitV <T> buat ( Lingkup lingkup, nilai Operand <T>, Operand <U> sizeSplits, sumbu Operand <Integer>, Long numSplit)
Metode pabrik untuk membuat kelas yang membungkus operasi SplitV baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
nilai | Tensor yang akan dipecah. |
ukuranSplit | daftar yang berisi ukuran setiap tensor keluaran sepanjang dimensi terpisah. Harus menjumlahkan dimensi nilai sepanjang split_dim. Dapat berisi satu -1 yang menunjukkan bahwa dimensi harus disimpulkan. |
sumbu | 0-D. Dimensi yang akan dipecah. Harus berada dalam rentang `[-peringkat(nilai), peringkat(nilai))`. |
Kembali
- contoh baru dari SplitV
Daftar publik< Keluaran <T>> keluaran ()
Tensor yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, yang ukurannya adalah `size_splits[i]`.