classe final pública EnqueueTPUEmbeddingBatch
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. |
---|