clase final pública SendTPUEmbeddingGradients
Realiza actualizaciones de degradado de tablas incrustadas.
Métodos públicos
static 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