PartitionedOutput

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>>
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

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

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.