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