SendTPUEmbeddingGradients

SendTPUEmbeddingGradients public final class

Realiza actualizaciones de degradado de tablas incrustadas.

Constantes

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

Métodos públicos

estáticas SendTPUEmbeddingGradients
crear ( Alcance alcance, Iterable < Operando < TFloat32 >> entradas, Iterable < Operando < TFloat32 >> learningRates, Cadena config)
Método de fábrica para crear una clase que envuelva una nueva operación SendTPUEmbeddingGradients.

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: "SendTPUEmbeddingGradients"

Métodos públicos

estáticas pública SendTPUEmbeddingGradients crean ( Alcance alcance, Iterable < operando < TFloat32 >> entradas, Iterable < operando < TFloat32 >> learningRates, Cadena config)

Método de fábrica para crear una clase que envuelva una nueva operación SendTPUEmbeddingGradients.

Parámetros
alcance alcance actual
entradas Una TensorList de degradados con la que actualizar las tablas de inserción. Este argumento tiene la misma longitud y formas que el valor de retorno de RecvTPUEmbeddingActivations, pero contiene gradientes de la pérdida del modelo con respecto a las activaciones de incrustación. Las tablas de incrustación se actualizan a partir de estos gradientes a través del optimizador especificado en la configuración de incrustación de TPU dada a tpu.initialize_system.
tasas de aprendizaje Una TensorList de escalares float32, uno para cada etiqueta de tasa de aprendizaje dinámico: vea los comentarios en //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Varias tablas pueden compartir la misma etiqueta de tasa de aprendizaje dinámico como se especifica en la configuración. Si las tasas de aprendizaje de todas las tablas son constantes, esta lista debería estar vacía.
config Protocolo TPUEmbeddingConfiguration serializado.
Devoluciones
  • una nueva instancia de SendTPUEmbeddingGradients