TPUPartitionedOutput

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

De la classe org.tensorflow.op.PrimitiveOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie l' Operation sous-jacente
Chaîne finale	àChaîne ()

De la classe java.lang.Object

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

Depuis l'interface java.lang.Iterable

vide	forEach (Consommateur <? super T> arg0)
Itérateur abstrait < Opérande <T>>	itérateur ()
Spliterator< Opérande <T>>	séparateur ()

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

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 TPUPartitionedOutput.Options partitionDim (Long partitionDim)

Paramètres

partitionDim	Un entier décrit quelle dimension est partitionnée.