classe finale publique PartitionedOutput
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 | PartitionedOutput.Options | Attributs facultatifs pour PartitionedOutput |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
statique <T étend TType > PartitionedOutput <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 PartitionedOutput. |
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. |
PartitionedOutput.Options statique | partitionDim (Longue partitionDim) |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Valeur constante : "TPUPartitionedOutput"
Méthodes publiques
public static PartitionedOutput <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 PartitionedOutput.
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 PartitionedOutput
liste publique< Sortie <T>> sortie ()
Une liste d'entrées partitionnées qui doivent avoir la même forme.
public statique PartitionedOutput.Options partitionDim (Long partitionDim)
Paramètres
partitionDim | Un entier décrit quelle dimension est partitionnée. |
---|