XlaSplitND

classe finale publique XlaSplitND

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]]
 
`num_splits` :
[2, 2]
 
et `paddings` :
[1, 1]
 
les "sorties" attendues sont :
[[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>>
Liste< Sortie <T>>
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
Retour
  • une nouvelle instance de XlaSplitND

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

liste publique < Sortie <T>> sorties ()

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.