InfeedEnqueueTuple classe final pública
Alimenta vários valores do Tensor no cálculo como uma tupla XLA.
Classes aninhadas
| classe | InfeedEnqueueTuple.Options | Atributos opcionais para InfeedEnqueueTuple | |
Métodos Públicos
| estática 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. |
| estáticos InfeedEnqueueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
| estáticos InfeedEnqueueTuple.Options | layouts (List <longo> layouts) |
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. |
|---|
public static InfeedEnqueueTuple.Options layouts (List <longo> layouts)
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. |
|---|