classe finale publique AssignVariableXlaConcatND
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 écrit le tenseur fusionné sans remplissage dans la variable de ressource.
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]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Classes imbriquées
| classe | AssignVariableXlaConcatND.Options | Attributs facultatifs pour AssignVariableXlaConcatND | |
Méthodes publiques
| statique <T> AssignVariableXlaConcatND | créer ( Scope scope, Operand <?> ressource, Iterable< Operand <T>> entrées, List<Long> numConcats, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération AssignVariableXlaConcatND. |
| AssignVariableXlaConcatND.Options statique | remplissages (List<Long> remplissages) |
Méthodes héritées
Méthodes publiques
public static AssignVariableXlaConcatND create ( Scope scope, Operand <?> ressource, Iterable< Operand <T>> entrées, List<Long> numConcats, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération AssignVariableXlaConcatND.
Paramètres
| portée | portée actuelle |
|---|---|
| Ressource | Variable de ressource pour les tenseurs d'entrée concaténés dans toutes les dimensions. } in_arg { nom : "entrées" description : < |
| numConcats | Nombre de façons de fusionner par dimension. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de AssignVariableXlaConcatND
remplissages publics statiques AssignVariableXlaConcatND.Options (remplissages 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. |
|---|