TPUEmbeddingActivations

TPUEmbeddingActivations classe final públicas

Uma operação que permite a diferenciação de embeddings TPU.

Esta operação simplesmente retorna sua primeira entrada, que se presume ter sido dividida dos tensores retornados por TPUEmbeddingDequeueActivations. A presença desta operação, e seu primeiro argumento sendo uma variável treinável, permite a diferenciação automática de gráficos contendo embeddings por meio das bibliotecas Python de incorporação de TPU.

Métodos Públicos

Output <float>
asOutput ()
Retorna o identificador simbólico de um tensor.
estáticos TPUEmbeddingActivations
criar ( Scope escopo, Operando <float> embeddingVariable, Operando <float> slicedActivations, Longo tableID, Long lookupId)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUEmbeddingActivations.
Output <float>
saída ()

Métodos herdados

Métodos Públicos

pública Output <float> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static TPUEmbeddingActivations criar ( Scope escopo, Operando <float> embeddingVariable, Operando <float> slicedActivations, Longo tableID, Long lookupId)

Método de fábrica para criar uma classe envolvendo uma nova operação TPUEmbeddingActivations.

Parâmetros
alcance escopo atual
embeddingVariable Uma variável treinável, permitindo que os otimizadores encontrem esta op.
SlicedActivations O Tensor de ativações de incorporação a retornar.
tableId O id da tabela na configuração da camada de incorporação a partir da qual essas ativações foram calculadas.
lookupId Identificador do conjunto de índices de incorporação que produziram essas ativações.
Devoluções
  • uma nova instância de TPUEmbeddingActivations

pública Output <float> saída ()