SplitV

publiczna klasa końcowa SplitV

Dzieli tensor na tensory `num_split` wzdłuż jednego wymiaru.

Metody publiczne

statyczny <T, U rozszerza Numer> SplitV <T>
utwórz ( Zakres zakresu, Wartość argumentu <T>, Operand <U> sizeSplits, Operand <Integer> oś, Long numSplit)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SplitV.
Iterator< Operand <T>>
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.
0-D. Wymiar, wzdłuż którego nastąpi podział. Musi należeć do zakresu `[-rank(wartość), ranga(wartość))`.
Zwroty
  • nowa instancja SplitV

publiczny Iterator< Operand <T>> iterator ()

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