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
Output <float> | asOutput () Retorna o identificador simbólico de um tensor. |
estáticos TPUEmbeddingActivations | |
Output <float> | saída () |
Métodos herdados
Métodos Públicos
pública 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 criar ( Scope escopo, Operando <float> embeddingVariable, Operando <float> slicedActivations, Longo 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