SendTPUEmbeddingGradients

clase final pública SendTPUEmbeddingGradients

Realiza actualizaciones de gradiente de tablas de incrustación.

Métodos públicos

SendTPUEmbeddingGradients estático
crear (alcance alcance , Iterable< Operando <Float>> entradas, Iterable< Operando <Float>> learningRates, 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

creación pública estática de SendTPUEmbeddingGradients (alcance del alcance , entradas Iterable < operando <flotante>>, 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.

Parámetros
alcance alcance actual
entradas Una TensorList de gradientes con los que actualizar tablas de incrustación. Este argumento tiene la misma longitud y forma 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 en tpu.initialize_system.
aprendizajeTarifas Una TensorList de escalares float32, uno para cada etiqueta de tasa de aprendizaje dinámico: consulte 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 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.
configuración Protocolo de configuración de integración de TPU serializado.
Devoluciones
  • una nueva instancia de SendTPUEmbeddingGradients