classe final pública EnqueueTPUEmbeddingBatch
Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Classes aninhadas
| aula | EnqueueTPUEmbeddingBatch.Options | Atributos opcionais para EnqueueTPUEmbeddingBatch | |
Métodos Públicos
| EnqueueTPUEmbeddingBatch.Options estático | combinadores (combinadores List<String>) |
| EnqueueTPUEmbeddingBatch estático | |
| EnqueueTPUEmbeddingBatch.Options estático | deviceOrdinal (dispositivo longo Ordinal) |
Métodos herdados
Métodos Públicos
combinadores public static EnqueueTPUEmbeddingBatch.Options (combinadores List<String>)
Parâmetros
| combinadores | Uma lista de escalares de cadeia de caracteres, um para cada tabela de incorporação que especifica como normalizar as ativações de incorporação após a soma ponderada. Os combinadores suportados são 'mean', 'sum' ou 'sqrtn'. É inválido que a soma dos pesos seja 0 para 'média' ou que a soma dos pesos quadrados seja 0 para 'sqrtn'. Se os combinadores não forem passados, o padrão é usar 'soma' para todas as tabelas. |
|---|
public static EnqueueTPUEmbeddingBatch create ( Escopo de escopo , Iterável< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingBatch.
Parâmetros
| escopo | escopo atual |
|---|---|
| lote | Uma lista de tensores 1D, um para cada tabela de incorporação, contendo as entradas em lote codificadas como protos dist_belief.SparseFeatures. Se o campo de peso no proto SparseFeatures não estiver preenchido para um ID, um peso de 1,0 será assumido. |
| 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 EnqueueTPUEmbeddingBatch
public static EnqueueTPUEmbeddingBatch.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. |
|---|