SendTPUEmbeddingGradients

clase final pública SendTPUEmbeddingGradients

Realiza actualizaciones de degradado de tablas incrustadas.

Métodos públicos

static SendTPUEmbeddingGradients
crear ( ámbito de alcance, Iterable< Operando <Flotante>> entradas, Iterable< Operando <Flotante>> tasas de aprendizaje, Configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación SendTPUEmbeddingGradients.

Métodos Heredados

Métodos públicos

Public static SendTPUEmbeddingGradients create ( Scope scope, Iterable< Operand <Float>> input, Iterable< Operand <Float>> learningRates, String config)

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

Parámetros
alcance alcance actual
entradas Una TensorList de degradados con los que actualizar las tablas de incrustación. Este argumento tiene la misma longitud y formas que el valor de retorno de RecvTPUEmbeddingActivations, pero contiene gradientes de 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 proporcionada a tpu.initialize_system.
tasas de aprendizaje Una TensorList de escalares float32, una para cada etiqueta de tasa de aprendizaje dinámica: 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ámica según lo especificado en la configuración. Si las tasas de aprendizaje de todas las tablas son constantes, esta lista debería estar vacía.
configuración Proto de configuración de integración de TPU serializado.
Devoluciones
  • una nueva instancia de SendTPUEmbeddingGradients