TPUEmbeddingActivations

classe final pública TPUEmbeddingActivations

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

Esta operação simplesmente retorna sua primeira entrada, que se presume ter sido cortada 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 TPU Embedding Python.

Métodos Públicos

Saída <flutuante>
asOutput ()
Retorna o identificador simbólico de um tensor.
TPUEmbeddingActivations estáticas
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 <flutuante>
saída ()

Métodos herdados

Métodos Públicos

Saída pública <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 de 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
escopo escopo atual
incorporaçãoVariável Uma variável treinável, permitindo que os otimizadores encontrem esta operação.
fatiadoAtivações A incorporação ativa o Tensor para retornar.
tabelaId O ID da tabela na configuração da camada de incorporação a partir da qual essas ativações foram calculadas.
ID de pesquisa 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> saída ()