classe final pública InfeedEnqueueTuple
Alimenta vários valores de Tensor no cálculo como uma tupla XLA.
Classes aninhadas
| aula | InfeedEnqueueTuple.Options | Atributos opcionais para InfeedEnqueueTuple | |
Métodos Públicos
| InfeedEnqueueTuple estático | |
| InfeedEnqueueTuple.Options estático | deviceOrdinal (dispositivo longo Ordinal) |
| InfeedEnqueueTuple.Options estático | layouts (lista <longo> layouts) |
Métodos herdados
Métodos Públicos
public static InfeedEnqueueTuple create ( Escopo do escopo , entradas Iterable< Operand <?>>, List< Shape > formas, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação InfeedEnqueueTuple.
Parâmetros
| escopo | 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 estiver sendo executado em um dispositivo TPU e >= 0 quando o Op estiver sendo executado no dispositivo CPU. |
|---|
layouts InfeedEnqueueTuple.Options estáticos públicos (layouts List<Long>)
Parâmetros
| layouts | Um vetor que contém o layout solicitado em sequência menor para maior para todas as formas da 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. |
|---|