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 ha sido cortada 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.
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Salida <TFloat32> | como salida () Devuelve el identificador simbólico del tensor. |
| Activaciones de incrustación de TPU estáticas | crear (alcance alcance , Operando <TFloat32> incrustaciónVariable, Operando <TFloat32> activaciones cortadas, Long tableId, Long lookupId) Método de fábrica para crear una clase que envuelve una nueva operación TPUEmbeddingActivations. |
| Salida <TFloat32> | producción () |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <TFloat32> asOutput ()
Devuelve el identificador simbólico del tensor.
Las entradas a 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.
creación de TPUEmbeddingActivations estáticas públicas (alcance del alcance , operando <TFloat32> embeddingVariable, operando <TFloat32> activaciones cortadas, Id de tabla larga, Id de búsqueda larga)
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. |
| activados en rodajas | La incrustación activa Tensor para regresar. |
| ID de tabla | La identificación de la tabla en la configuración de la capa de incrustación 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