TPUReplicatedOutput

classe finale publique TPUReplicatedOutput

Connecte N sorties d’un calcul TPU répliqué à N voies.

Cette opération contient une sortie répliquée d'un sous-graphe de calcul `tpu.replicate()`. Chaque sortie répliquée a la même forme et le même type à côté de l'entrée.

Par exemple :

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 
Le calcul ci-dessus a une sortie répliquée de deux répliques.

Méthodes publiques

statique <T> TPUReplicatedOutput <T>
créer (portée de portée , entrée d'opérande <T>, numReplicas longs)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUReplicatedOutput.
Itérateur< Opérande <T>>
Liste< Sortie <T>>

Méthodes héritées

Méthodes publiques

public static TPUReplicatedOutput <T> créer (portée de portée , entrée d'opérande <T>, numReplicas longs)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUReplicatedOutput.

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de TPUReplicatedOutput

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

liste publique < Sortie <T>> sorties ()