SplitV

Classe finale publique SplitV

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

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

statique <T étend TType > SplitV <T>
créer ( Portée de la portée , valeur de l'opérande <T>, opérande <? extends TNumber > sizeSplits, opérande < TInt32 > axe, 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

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "SplitV"

Méthodes publiques

public static SplitV <T> créer (portée de portée , valeur de l'opérande <T>, opérande <? extends TNumber > sizeSplits, opérande < TInt32 > axe, 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] ».