InfeedEnqueueTuple

public clase final InfeedEnqueueTuple

Introduce varios valores de Tensor en el cálculo como una tupla XLA.

Clases anidadas

clase InfeedEnqueueTuple.Options Atributos opcionales para InfeedEnqueueTuple

Métodos públicos

estático InfeedEnqueueTuple
crear ( ámbito de alcance, Iterable < operando <? >> entradas, lista < forma > formas, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueueTuple.
static InfeedEnqueueTuple.Options
deviceOrdinal (Long deviceOrdinal)
static InfeedEnqueueTuple.Options
diseños (Lista de diseños <Long>)

Métodos heredados

Métodos públicos

public static InfeedEnqueueTuple create ( Alcance del alcance, Iterable < Operand <? >> entradas, List < Shape > 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 utilizando 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.

diseños de InfeedEnqueueTuple.Options estáticos públicos ( diseños 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.