InfeedEnqueueTuple

classe final pública InfeedEnqueueTuple

Alimenta vários valores do Tensor no cálculo como uma tupla XLA.

Classes aninhadas

aula InfeedEnqueueTuple.Options Atributos opcionais para InfeedEnqueueTuple

Métodos Públicos

static InfeedEnqueueTuple
criar ( escopo do escopo, entradas de < operandos <? >> iteráveis, formas de lista < forma >, opções ... opções)
Método de fábrica para criar uma classe que envolve uma nova operação InfeedEnqueueTuple.
static InfeedEnqueueTuple.Options
deviceOrdinal (Long deviceOrdinal)
static InfeedEnqueueTuple.Options
layouts (Listar layouts <Long>)

Métodos herdados

Métodos Públicos

public static InfeedEnqueueTuple criar ( Scope escopo, Iterable < Operando <? >> entradas, List < Forma > formas, Options ... Opções)

Método de fábrica para criar uma classe que envolve 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
deviceOrdinal O dispositivo TPU a ser usado. Deve ser -1 quando o Op está em execução em um dispositivo TPU e> = 0 quando o Op está em execução no dispositivo de CPU.

layouts public static InfeedEnqueueTuple.Options ( layouts de lista <Long>)

Parâmetros
layouts Um vetor que mantém o layout solicitado em sequência de menor para 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.