InfeedEnqueueTuple

classe final pública InfeedEnqueueTuple

Alimenta vários valores de tensor na computação como uma tupla XLA.

Classes aninhadas

aula InfeedEnqueueTuple.Options Atributos opcionais para InfeedEnqueueTuple

Métodos públicos

static InfeedEnqueueTuple
create ( Escopo do escopo, Iterable< Operando <?>> entradas, List< Shape > shapes, Options... options)
Método de fábrica para criar uma classe envolvendo uma nova operação InfeedEnqueueTuple.
static InfeedEnqueueTuple.Options
deviceOrdinal (Longo deviceOrdinal)
static InfeedEnqueueTuple.Options
layouts (List<Long> layouts)

Métodos Herdados

Métodos públicos

public static InfeedEnqueueTuple create ( Escopo do escopo, Iterable< Operando <?>> entradas, List< Shape > shapes, Options... options)

Método de fábrica para criar uma classe envolvendo uma nova operação InfeedEnqueueTuple.

Parâmetros
alcance escopo atual
entradas Uma lista de tensores que serão fornecidos usando o mecanismo de alimentação.
formas As formas de cada tensor em `entradas`.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de InfeedEnqueueTuple

public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parâmetros
dispositivoOrdinal O dispositivo TPU a ser usado. Deve ser -1 quando o Op está sendo executado em um dispositivo TPU e >= 0 quando o Op está sendo executado no dispositivo CPU.

Public static InfeedEnqueueTuple.Options layouts (List<Long> layouts)

Parâmetros
layouts Um vetor que mantém o layout solicitado na sequência menor a maior para todas as formas de tupla, na ordem em que as formas aparecem na entrada "formas". Os elementos de layout para uma subforma podem ser definidos como -1, caso em que o layout correspondente será calculado pela operação de alimentação.