Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Um op 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
static EnqueueTPUEmbeddingBatch.Options | combinadores (List<String> combinadores) |
static EnqueueTPUEmbeddingBatch | |
static EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (Longo deviceOrdinal) |
Métodos Herdados
Métodos públicos
public static EnqueueTPUEmbeddingBatch.Options combinadores (List<String> combinadores)
Parâmetros
combinadores | Uma lista de escalares de string, 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 a soma dos pesos quadrados seja 0 para 'sqrtn'. Se os combinadores não forem passados, o padrão é usar 'sum' para todas as tabelas. |
---|
public static EnqueueTPUEmbeddingBatch create ( Escopo do escopo, Iterable< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)
Método de fábrica para criar uma classe envolvendo uma nova operação EnqueueTPUEmbeddingBatch.
Parâmetros
alcance | escopo atual |
---|---|
lote | Uma lista de tensores 1D, um para cada tabela de incorporação, contendo as entradas de lote codificadas como protos dist_belief.SparseFeatures. Se o campo de peso no proto SparseFeatures não for preenchido para um ID, um peso de 1,0 será assumido. |
modeOverride | Uma entrada de string que substitui o modo especificado no TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando definido como 'unspecified', 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 está sendo executado em um dispositivo TPU e >= 0 quando o Op está sendo executado no dispositivo CPU. |
---|