Una operación que permite la diferenciación de las incrustaciones de TPU.
Esta operación simplemente devuelve su primera entrada, que se supone que se ha separado de los tensores devueltos por TPUEmbeddingDequeueActivations. La presencia de esta operación, y su primer argumento es una variable entrenable, permite la diferenciación automática de gráficos que contienen incrustaciones a través de las bibliotecas TPU Embedding Python.
Métodos públicos
Salida <Flotante> | como salida () Devuelve el identificador simbólico de un tensor. |
Activaciones de inserción de TPU estáticas | |
Salida <Flotante> | salida () |
Métodos Heredados
Métodos públicos
Salida pública <Flotante> asOutput ()
Devuelve el identificador simbólico de un tensor.
Las entradas de las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
TPUEmbeddingActivations estático público crear ( Scope scope, Operand <Float> embeddingVariable, Operand <Float> slicedActivations, Long tableId, Long lookupId)
Método de fábrica para crear una clase que envuelve una nueva operación TPUEmbeddingActivations.
Parámetros
alcance | alcance actual |
---|---|
incrustaciónVariable | Una variable entrenable, que permite a los optimizadores encontrar esta operación. |
Activaciones en rodajas | Las activaciones de incrustación Tensor para volver. |
ID de tabla | El id de la tabla en la configuración de la capa incrustada a partir de la cual se calcularon estas activaciones. |
ID de búsqueda | Identificador del conjunto de índices de incrustación que produjeron estas activaciones. |
Devoluciones
- una nueva instancia de TPUEmbeddingActivations