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 | |
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