clase final pública OutfeedDequeueTuple
Recupere varios valores de la salida de cálculo.
Esta operación se bloqueará indefinidamente hasta que los datos estén disponibles. La salida `i` corresponde al elemento de tupla de XLA` i`.
Clases anidadas
clase | OutfeedDequeueTuple.Options | Atributos opcionales para OutfeedDequeueTuple |
Métodos públicos
static OutfeedDequeueTuple | crear ( ámbito de alcance, List <Class <? >> dtypes, List < Shape > formas, Opciones ... opciones) Método de fábrica para crear una clase que envuelva una nueva operación OutfeedDequeueTuple. |
static OutfeedDequeueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
Iterador < Operando <Objeto>> | iterador () |
Lista < Salida <? >> | salidas () Una lista de tensores que se leerán de la salida. |
Métodos heredados
Métodos públicos
public static OutfeedDequeueTuple create ( Alcance del alcance, List <Class <? >> dtypes, List < Shape > formas, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación OutfeedDequeueTuple.
Parámetros
alcance | alcance actual |
---|---|
dtipos | Los tipos de elementos de cada elemento en "salidas". |
formas | Las formas de cada tensor en "salidas". |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de OutfeedDequeueTuple
public static OutfeedDequeueTuple.Options deviceOrdinal (Long deviceOrdinal)
Parámetros
deviceOrdinal | El dispositivo de TPU que se utilizará. Debe ser -1 cuando el Op se está ejecutando en un dispositivo de TPU y> = 0 cuando el Op se está ejecutando en el dispositivo de la CPU. |
---|