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. |
---|