XlaConcatND

classe finale publique XlaConcatND

Concatène le tenseur d'entrée dans toutes les dimensions.

Une opération qui fusionne tranche le tenseur d'entrée en fonction de l'attribut num_splits donné, supprime éventuellement les remplissages et renvoie le tenseur fusionné sans remplissages.

Cette opération peut être générée via le pont TPU.

Par exemple, avec le tenseur `input` : 

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
`num_splits` : 
[2, 2]
et `rembourrages` : 
[1, 1]
les « résultats » attendus sont : 
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]

Classes imbriquées

classe XlaConcatND.Options Attributs facultatifs pour XlaConcatND

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T> XlaConcatND <T>
créer ( Scope scope, Iterable< Operand <T>> entrées, List<Long> numConcats, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaConcatND.
Sortie <T>
sortir ()
statique XlaConcatND.Options
remplissages (List<Long> remplissages)

Méthodes héritées

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique d'un tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static XlaConcatND <T> create ( Scope scope, Iterable< Operand <T>> entrées, List<Long> numConcats, Options... options)

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

Paramètres
portée portée actuelle
entrées Saisissez les tranches de tenseur dans l'ordre des lignes principales pour fusionner dans toutes les dimensions. Toutes les entrées doivent avoir la même forme. } out_arg { nom : "sortie" description : <
numConcats Nombre de façons de fusionner par dimension.
choix porte des valeurs d'attributs facultatifs
Retours
  • une nouvelle instance de XlaConcatND

sortie publique <T> sortie ()

remplissages publics statiques XlaConcatND.Options (rembourrages List<Long>)

Paramètres
rembourrages Liste facultative de remplissages droits par dimension à supprimer du tenseur fusionné final. Ces remplissages ne doivent pas dépasser la dimension du résultat fusionné avant de supprimer les remplissages.