classe final pública EnqueueTPUEmbeddingIntegerBatch
Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Classes aninhadas
aula | EnqueueTPUEmbeddingIntegerBatch.Options | Atributos opcionais para EnqueueTPUEmbeddingIntegerBatch |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático EnqueueTPUEmbeddingIntegerBatch | |
estático EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (dispositivo longo Ordinal) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Valor constante: "EnqueueTPUEmbeddingIntegerBatch"
Métodos Públicos
public static EnqueueTPUEmbeddingIntegerBatch create ( Scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > modeOverride, Options... options)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingIntegerBatch.
Parâmetros
escopo | escopo atual |
---|---|
lote | Uma lista de tensores 1D, um para cada tabela incorporada, contendo os índices nas tabelas. |
modoSubstituir | Uma entrada de string que substitui o modo especificado em TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando definido como 'não especificado', o modo definido em TPUEmbeddingConfiguration é usado, caso contrário, mode_override é usado. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Parâmetros
dispositivoOrdinal | O dispositivo TPU a ser usado. Deve ser >= 0 e menor que o número de núcleos de TPU na tarefa na qual o nó está colocado. |
---|