Split

Classe finale publique Split

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

Méthodes publiques

statique <T> Fractionné <T>
créer (portée de portée , axe de l'opérande <Integer>, valeur de l'opérande <T>, Long numSplit)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération Split.
Itérateur< Opérande <T>>
itérateur ()
Liste< Sortie <T>>
sortir ()
Ce sont des tenseurs de forme identique, dont la forme correspond à celle de « valeur » sauf le long de « l'axe », où leurs tailles sont « valeurs.shape[split_dim] / num_split ».

Méthodes héritées

Méthodes publiques

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

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

Paramètres
portée portée actuelle
axe 0-D. Dimension selon laquelle diviser. Doit être dans la plage `[-rank(value), Rank(value))`.
valeur Le tenseur à diviser.
numSplit Le nombre de façons de diviser. Doit diviser uniformément `value.shape[split_dim]`.
Retour
  • une nouvelle instance de Split

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

liste publique< Sortie <T>> sortie ()

Ce sont des tenseurs de forme identique, dont la forme correspond à celle de « valeur » sauf le long de « l'axe », où leurs tailles sont « valeurs.shape[split_dim] / num_split ».