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.
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| Saída < TFloat32 > | asOutput () Retorna o identificador simbólico do tensor. |
| EmbeddingActivations estáticas | |
| Saída < TFloat32 > | saída () |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública < TFloat32 > asOutput ()
Retorna o identificador simbólico do 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 EmbeddingActivations create ( Escopo de escopo , Operando < TFloat32 > embeddingVariable, Operando < TFloat32 > slicedActivations, Long tableId, Long lookupId)
Método de fábrica para criar uma classe que envolve uma nova operação EmbeddingActivations.
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 EmbeddingActivations