TPUEmbeddingActivations

public final class TPUEmbeddingActivations

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

Resultado <Float>
asOutput ()
Retorna o identificador simbólico de um tensor.
TPUEmbeddingActivations estáticas
criar ( escopo do escopo, Operand <Float> embeddingVariable, Operand <Float> SlicedActivations, Long tableId, Long lookupId)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUEmbeddingActivations.
Resultado <Float>
saída ()

Métodos herdados

Métodos Públicos

public 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 create ( escopo do escopo, Operand <Float> embeddingVariable, Operand <Float> sliceedActivations, 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

Saída pública <Float> output ()