InfeedEnqueueTuple public final class
Introduce varios valores de tensor en el cálculo como una tupla XLA.
Clases anidadas
| clase | InfeedEnqueueTuple.Options | Los atributos opcionales para InfeedEnqueueTuple | |
Métodos públicos
| estática InfeedEnqueueTuple | crear ( Alcance alcance, Iterable < operando <? >> entradas, List < Forma > formas, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueueTuple. |
| estáticas InfeedEnqueueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
| estáticas InfeedEnqueueTuple.Options | diseños (List <Larga> diseños) |
Métodos heredados
Métodos públicos
estáticas pública InfeedEnqueueTuple crean ( Alcance alcance, Iterable < operando <? >> entradas, List < Forma > formas, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva 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
public static InfeedEnqueueTuple.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. |
|---|
públicas estáticas InfeedEnqueueTuple.Options diseños (List <Larga> diseños)
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 las formas 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. |
|---|