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