Divise le tenseur d'entrée des variables de ressource sur toutes les dimensions.
Une opération qui divise le tenseur d'entrée de la variable de ressource en fonction de l'attribut num_splits donné, remplit éventuellement les tranches 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]]
`num_splits` : [2, 2]
et `rembourrages` : [1, 1]
les « résultats » attendus sont : [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Classes imbriquées
classe | ReadVariableXlaSplitND.Options | Attributs facultatifs pour ReadVariableXlaSplitND |
Méthodes publiques
statique <T> ReadVariableXlaSplitND <T> | créer ( Scope scope, Operand <?> ressource, Class<T> T, Long N, List<Long> numSplits, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ReadVariableXlaSplitND. |
Itérateur< Opérande <T>> | itérateur () |
Liste< Sortie <T>> | sorties () |
statique ReadVariableXlaSplitND.Options | remplissages (List<Long> remplissages) |
Méthodes héritées
Méthodes publiques
public static ReadVariableXlaSplitND <T> create ( Scope scope, Operand <?> ressource, Class<T> T, Long N, List<Long> numSplits, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ReadVariableXlaSplitND.
Paramètres
portée | portée actuelle |
---|---|
ressource | Variable de ressource du 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 ReadVariableXlaSplitND
remplissages publics statiques ReadVariableXlaSplitND.Options (remplissages 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. |
---|