Un sélecteur de noyau TPU Op.
Cette opération produit un ensemble de cœurs TPU (pour l'échauffement) ou un seul cœur TPU (pour une inférence régulière) sur lequel exécuter le programme TPU. La sortie est consommée par TPUPartitionedCall.
Méthodes publiques
Sortie <Entier> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
TPUOrdinalSelector statique | créer (portée d'application ) Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUOrdinalSelector. |
Sortie <Entier> | appareilsOrdinaux () Un vecteur d’un ou plusieurs cœurs TPU. |
Méthodes héritées
Méthodes publiques
sortie publique <Integer> asOutput ()
Renvoie le handle symbolique d'un tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static TPUOrdinalSelector créer (portée de portée )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUOrdinalSelector.
Paramètres
portée | portée actuelle |
---|
Retour
- une nouvelle instance de TPUOrdinalSelector