classe finale publique InfeedEnqueueTuple
Introduire plusieurs valeurs Tensor dans le calcul sous forme de tuple XLA.
Classes imbriquées
| classe | InfeedEnqueueTuple.Options | Attributs facultatifs pour InfeedEnqueueTuple | |
Méthodes publiques
| InfeedEnqueueTuple statique | créer (portée de portée , entrées Iterable< Operand <?>>, formes List< Shape >, options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération InfeedEnqueueTuple. |
| statique InfeedEnqueueTuple.Options | appareilOrdinal (appareilOrdinal long) |
| statique InfeedEnqueueTuple.Options | mises en page (mises en page List<Long>) |
Méthodes héritées
Méthodes publiques
public static InfeedEnqueueTuple create ( Scope scope, Iterable< Operand <?>> entrées, List< Shape > formes, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération InfeedEnqueueTuple.
Paramètres
| portée | portée actuelle |
|---|---|
| contributions | Une liste de tenseurs qui seront fournis à l'aide du mécanisme d'alimentation. |
| formes | Les formes de chaque tenseur dans les « entrées ». |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de InfeedEnqueueTuple
public statique InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
Paramètres
| appareilOrdinal | 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. |
|---|
Dispositions statiques publiques InfeedEnqueueTuple.Options (mises en page List<Long>)
Paramètres
| mises en page | Un vecteur contenant la disposition demandée dans une séquence mineure à majeure pour toutes les formes de tuple, dans l'ordre dans lequel les formes apparaissent dans l'entrée "formes". Les éléments de présentation d'une sous-forme peuvent être définis sur -1, auquel cas la présentation correspondante sera calculée par l'opération d'alimentation. |
|---|