InfeedEnqueueTuple

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.