clase final pública InfeedEnqueueTuple
Alimenta varios valores de tensor en el cálculo como una tupla XLA.
Clases anidadas
clase | InfeedEnqueueTuple.Opciones | Atributos opcionales para InfeedEnqueueTuple |
Métodos públicos
InfeedEnqueueTuple estático | crear ( Ámbito de 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. |
static InfeedEnqueueTuple.Options | deviceOrdinal (dispositivo largo ordinal) |
static InfeedEnqueueTuple.Options | diseños (Lista <Long> diseños) |
Métodos Heredados
Métodos públicos
public static InfeedEnqueueTuple create ( Scope scope, Iterable< Operand <?>> entradas, List< Shape > formas, 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 utilizando el mecanismo de alimentación. |
formas | Las formas de cada tensor en `inputs`. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de InfeedEnqueueTuple
public static InfeedEnqueueTuple.Options dispositivo ordinal ( dispositivo largo ordinal)
Parámetros
dispositivoordinal | El dispositivo de TPU a utilizar. Debe ser -1 cuando la operación se ejecuta en un dispositivo de TPU y >= 0 cuando la operación se ejecuta en el dispositivo de CPU. |
---|
Diseños de InfeedEnqueueTuple.Options estáticos públicos ( diseños de List<Long>)
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. |
---|