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>> | itérateur () |
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
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] ».