clase final pública SendTPUEmbeddingGradients
Realiza actualizaciones de gradiente de tablas de incrustación.
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| SendTPUEmbeddingGradients estático |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Valor constante: "SendTPUEmbeddingGradients"
Métodos públicos
creación pública estática de SendTPUEmbeddingGradients (alcance de alcance , Iterable < operando < TFloat32 >> entradas, iterable < operando < TFloat32 >> learningRates, 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