ReadVariableXlaSplitND

classe finale publique ReadVariableXlaSplitND

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

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

liste publique < Sortie <T>> sorties ()

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.