clase final pública InfeedEnqueueTuple
Introduce múltiples valores de tensor en el cálculo como una tupla XLA.
Clases anidadas
| clase | InfeedEnqueueTuple.Opciones | Atributos opcionales para InfeedEnqueueTuple | |
Métodos públicos
| Entrada estáticaEnqueueTuple | crear ( alcance alcance, Iterable< Operando <?>> entradas, Lista< Forma > formas, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación InfeedEnqueueTuple. |
| InfeedEnqueueTuple.Options estático | dispositivoOrdinal (dispositivo largoOrdinal) |
| InfeedEnqueueTuple.Options estático | diseños (Lista<Largo> diseños) |
Métodos heredados
Métodos públicos
creación pública estática de InfeedEnqueueTuple (alcance de alcance , entradas Iterable< operando <?>>, formas de lista <forma> , opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación InfeedEnqueueTuple.
Parámetros
| alcance | alcance actual |
|---|---|
| entradas | Una lista de tensores que se proporcionarán mediante el mecanismo de alimentación. |
| formas | Las formas de cada tensor en "entradas". |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de InfeedEnqueueTuple
público estático InfeedEnqueueTuple.Options dispositivoOrdinal (dispositivo largo Ordinal)
Parámetros
| dispositivoOrdinal | El dispositivo TPU a utilizar. Esto debe ser -1 cuando el Op se ejecuta en un dispositivo TPU y >= 0 cuando el Op se ejecuta en el dispositivo CPU. |
|---|
diseños públicos estáticos InfeedEnqueueTuple.Options (diseños de lista <largo>)
Parámetros
| diseños | Un vector que contiene el diseño solicitado en secuencia de menor a mayor para todas las formas de tupla, en el orden en que aparecen en la entrada "formas". Los elementos de diseño para una subforma se pueden establecer en -1, en cuyo caso la operación de alimentación calculará el diseño correspondiente. |
|---|