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 | |
Itérateur< Opérande <Objet>> | itérateur () |
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