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.

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

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

Métodos herdados

Constantes

nome_op final String public static

O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Valor constante: "TPUEmbeddingActivations"

Métodos Públicos

pública Output < TFloat32 > asOutput ()

Retorna o identificador simbólico do 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 < TFloat32 > embeddingVariable, Operando < TFloat32 > slicedActivations, Long 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 < TFloat32 > saída ()