Divise le tenseur d'entrée sur toutes les dimensions.
Une opération qui découpe le tenseur d'entrée en fonction de l'attribut num_splits donné, remplit les tranches en option et renvoie les tranches. Les tranches sont renvoyées dans l’ordre des lignes principales.
Cette opération peut être générée via le pont TPU.
Par exemple, avec le tenseur `input` :
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Classes imbriquées
classe | XlaSplitND.Options | Attributs facultatifs pour XlaSplitND |
Méthodes publiques
statique <T> XlaSplitND <T> | créer ( Scope scope, Operand <T> input, Long N, List<Long> numSplits, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSplitND. |
Itérateur< Opérande <T>> | itérateur () |
Liste< Sortie <T>> | sorties () |
XlaSplitND.Options statique | remplissages (List<Long> remplissages) |
Méthodes héritées
Méthodes publiques
public static XlaSplitND <T> créer (portée de portée , entrée d'opérande <T>, N long, liste <Long> numSplits, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSplitND.
Paramètres
portée | portée actuelle |
---|---|
saisir | Tenseur d'entrée à répartir sur toutes les dimensions. } out_arg { nom : "sorties" description : < |
numSplits | Nombre de façons de diviser par dimension. Les dimensions de la forme doivent être divisibles uniformément. |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de XlaSplitND
remplissages XlaSplitND.Options statiques publics (rembourrages List<Long>)
Paramètres
rembourrages | Liste facultative de remplissages droits par dimension du tenseur d'entrée à appliquer avant le fractionnement. Cela peut être utilisé pour rendre une dimension divisible de manière égale. |
---|