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 |
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