TPUEmbeddingActivations

TPUEmbeddingActivations public final class

Una operación que permite diferenciar 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 de incrustaciones de Python de TPU.

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

Salida < TFloat32 >
asOutput ()
Devuelve el identificador simbólico del tensor.
estáticas TPUEmbeddingActivations
crear ( Alcance alcance, operando < TFloat32 > embeddingVariable, operando < TFloat32 > slicedActivations, Long TableID largo lookupId)
Método de fábrica para crear una clase que envuelva una nueva operación TPUEmbeddingActivations.
Salida < TFloat32 >
salida ()

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "TPUEmbeddingActivations"

Métodos públicos

pública de salida < 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.

estáticas pública TPUEmbeddingActivations crean ( Alcance alcance, operando < TFloat32 > embeddingVariable, operando < TFloat32 > slicedActivations, Long TableID largo lookupId)

Método de fábrica para crear una clase que envuelva 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.
tableId El id de la tabla en la configuración de la capa de incrustación a partir de la cual se calcularon estas activaciones.
lookupId Identificador del conjunto de índices de inclusión que produjeron estas activaciones.
Devoluciones
  • una nueva instancia de TPUEmbeddingActivations

pública de salida < TFloat32 > salida ()