OutfeedDequeueTupleV2

classe finale publique OutfeedDequeueTupleV2

Récupérez plusieurs valeurs de la sortie de calcul. L'ordinal du périphérique est un tenseur permettant une sortie dynamique.

Cette opération se bloquera indéfiniment jusqu'à ce que les données soient disponibles. La sortie `i` correspond à l'élément de tuple XLA `i`.

Méthodes publiques

statique OutfeedDequeueTupleV2
créer ( Scope scope, Operand <Integer> deviceOrdinal, List<Class<?>> dtypes, List< Shape > formes)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération OutfeedDequeueTupleV2.
Itérateur< Opérande <Objet>>
Liste< Sortie <?>>
les sorties ()
Une liste de tenseurs qui seront lus à partir de la sortie.

Méthodes héritées

Méthodes publiques

public static OutfeedDequeueTupleV2 créer ( Scope scope, Operand <Integer> deviceOrdinal, List<Class<?>> dtypes, List< Shape > formes)

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

Paramètres
portée portée actuelle
appareilOrdinal Un tenseur scalaire int, représentant le périphérique TPU à utiliser. Cela doit être -1 lorsque l'opération s'exécute sur un périphérique TPU et >= 0 lorsque l'opération s'exécute sur le périphérique CPU.
types Les types d'éléments de chaque élément dans les « sorties ».
formes Les formes de chaque tenseur dans les « sorties ».
Retour
  • une nouvelle instance de OutfeedDequeueTupleV2

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

liste publique< Sortie <?>> sorties ()

Une liste de tenseurs qui seront lus à partir de la sortie.