public class akhir SplitV
Membagi tensor menjadi `num_split` tensor sepanjang satu dimensi.
Metode Publik
statis <T, U meluas Nomor> SplitV <T> | |
Iterator < Operan <T >> | iterator () |
Daftar < Keluaran <T >> | Output () Tensor yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, yang ukurannya adalah `pemisahan_ukuran[i]`. |
Metode yang Diwarisi
Metode Publik
public static SplitV <T> membuat ( Lingkup lingkup, Operan <T> nilai, Operan <U> sizeSplits, Operan <Integer> axis, Long numSplit)
Metode pabrik untuk membuat kelas yang membungkus operasi SplitV baru.
Parameter
cakupan | lingkup saat ini |
---|---|
nilai | Tensor untuk membelah. |
ukuranSplits | list yang berisi ukuran setiap tensor keluaran di sepanjang dimensi split. Harus menjumlahkan dimensi nilai di sepanjang split_dim. Dapat berisi satu -1 yang menunjukkan bahwa dimensi akan disimpulkan. |
sumbu | 0-D. Dimensi yang akan dipecah. Harus dalam rentang `[-rank(value), rank(value))`. |
Kembali
- contoh baru dari SplitV
public List < Keluaran <T >> keluaran ()
Tensor yang bentuknya cocok dengan `nilai` kecuali di sepanjang `sumbu`, yang ukurannya adalah `pemisahan_ukuran[i]`.