Dzieli tensor na tensory `num_split` wzdłuż jednego wymiaru.
Metody publiczne
statyczny <T, U rozszerza Numer> SplitV <T> | |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjście () Tensory, których kształt odpowiada kształtowi „wartość”, z wyjątkiem wzdłuż „osi”, gdzie ich rozmiary wynoszą „rozdział_rozmiaru[i]”. |
Metody dziedziczone
Metody publiczne
public static SplitV <T> utwórz ( Zakres zakresu, Wartość argumentu <T>, Argument <U> sizeSplits, Oś argumentu <Integer>, Long numSplit)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SplitV.
Parametry
zakres | aktualny zakres |
---|---|
wartość | Tensor do podziału. |
rozmiarPodziały | lista zawierająca rozmiary każdego tensora wyjściowego wzdłuż podzielonego wymiaru. Należy sumować do wymiaru wartości wzdłuż split_dim. Może zawierać jeden -1 wskazujący, że należy wywnioskować wymiar. |
oś | 0-D. Wymiar, wzdłuż którego nastąpi podział. Musi należeć do zakresu `[-rank(wartość), ranga(wartość))`. |
Zwroty
- nowa instancja SplitV
publiczna Lista< Wyjście <T>> wyjście ()
Tensory, których kształt odpowiada kształtowi „wartość”, z wyjątkiem wzdłuż „osi”, gdzie ich rozmiary wynoszą „rozdział_rozmiaru[i]”.