SplitV

Classe finale publique SplitV

Divise un tenseur en tenseurs `num_split` le long d'une dimension.

Méthodes publiques

statique <T, U étend le nombre> SplitV <T>
créer ( Portée de la portée , valeur de l'opérande <T>, opérande <U> sizeSplits, axe de l'opérande <Integer>, Long numSplit)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SplitV.
Itérateur< Opérande <T>>
Liste< Sortie <T>>
sortir ()
Tenseurs dont la forme correspond à celle de « valeur » sauf le long de « l'axe », où leurs tailles sont « size_splits[i] ».

Méthodes héritées

Méthodes publiques

public static SplitV <T> créer (portée de portée , valeur de l'opérande <T>, opérande <U> sizeSplits, axe de l' opérande <Integer>, Long numSplit)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération SplitV.

Paramètres
portée portée actuelle
valeur Le tenseur à diviser.
tailleSplits liste contenant les tailles de chaque tenseur de sortie le long de la dimension divisée. La somme doit correspondre à la dimension de la valeur le long de split_dim. Peut contenir un -1 indiquant que la dimension doit être déduite.
axe 0-D. Dimension selon laquelle diviser. Doit être dans la plage `[-rank(value), Rank(value))`.
Retour
  • une nouvelle instance de SplitV

public Iterator< Opérande <T>> itérateur ()

liste publique< Sortie <T>> sortie ()

Tenseurs dont la forme correspond à celle de « valeur » sauf le long de « l'axe », où leurs tailles sont « size_splits[i] ».