TPUEmbeddingActivations

classe final pública TPUEmbeddingActivations

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

Esta operação simplesmente retorna sua primeira entrada, que se supõe ter sido fatiada dos tensores retornados por TPUEmbeddingDequeueActivations. A presença deste op, 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 TPU Embedding Python.

Métodos públicos

Saída <Float>
comoSaída ()
Retorna o identificador simbólico de um tensor.
static TPUEmbeddingAtivações
create ( Escopo do escopo, Operando <Float> embeddingVariable, Operando <Float> slicedActivations, Long tableId, Long lookupId)
Método de fábrica para criar uma classe que envolve uma nova operação TPUEmbeddingActivations.
Saída <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 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, Operando <Float> embeddingVariable, Operando <Float> slicedActivations, Long tableId, Long lookupId)

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

Parâmetros
alcance escopo atual
embeddingVariable Uma variável treinável, permitindo que os otimizadores encontrem essa opção.
slicedAtivações O Tensor de ativação de incorporação para 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 embedding que produziram essas ativações.
Devoluções
  • uma nova instância de TPUEmbeddingActivations

saída pública <Float> saída ()