classe finale publique TPUPartitionedOutput
Une opération qui démultiplexe un tenseur à partitionner par XLA en une liste de partitionnés
sorties en dehors du calcul XLA.
Classes imbriquées
| classe | TPUPartitionedOutput.Options | Attributs facultatifs pour TPUPartitionedOutput | |
Méthodes publiques
| statique <T> TPUPartitionedOutput <T> | créer (portée de portée , entrées d'opérande <T>, options Long numSplits, Options... ) Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUPartitionedOutput. |
| Itérateur< Opérande <T>> | itérateur () |
| Liste< Sortie <T>> | sortir () Une liste d'entrées partitionnées qui doivent avoir la même forme. |
| TPUPartitionedOutput.Options statique | partitionDim (Longue partitionDim) |
Méthodes héritées
Méthodes publiques
public static TPUPartitionedOutput <T> create ( Scope scope, Operand <T> inputs, Long numSplits, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUPartitionedOutput.
Paramètres
| portée | portée actuelle |
|---|---|
| contributions | Un tenseur qui représente la forme complète des tenseurs partitionnés. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de TPUPartitionedOutput
liste publique< Sortie <T>> sortie ()
Une liste d'entrées partitionnées qui doivent avoir la même forme.
public statique TPUPartitionedOutput.Options partitionDim (Long partitionDim)
Paramètres
| partitionDim | Un entier décrit quelle dimension est partitionnée. |
|---|