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

org.tensorflow.op.RawOp
booléen final
est égal (Obj objet)
entier final
Code de hachage ()
Opération
op ()
Renvoie cette unité de calcul sous la forme d'une seule Operation .
Chaîne finale
àChaîne ()
booléen
est égal à (Objet arg0)
Classe finale<?>
obtenirClasse ()
int
Code de hachage ()
vide final
notifier ()
vide final
notifierTous ()
Chaîne
àChaîne ()
vide final
attendre (long arg0, int arg1)
vide final
attendez (long arg0)
vide final
attendez ()
org.tensorflow.op.Op
résuméExécutionEnvironnement
env ()
Renvoie l'environnement d'exécution dans lequel cette opération a été créée.
opération abstraite
op ()
Renvoie cette unité de calcul sous la forme d'une seule Operation .
vide abstrait
forEach (Consommateur <? super T> arg0)
Itérateur abstrait < Opérande <T>>
itérateur ()
Spliterator abstrait < Opérande <T>>
séparateur ()

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.