classe final pública SendTPUEmbeddingGradients
Executa atualizações de gradiente de tabelas incorporadas.
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| SendTPUEmbeddingGradients estáticos |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Valor constante: "SendTPUEmbeddingGradients"
Métodos Públicos
public static SendTPUEmbeddingGradients create ( Escopo de escopo , Iterable< Operand < TFloat32 >> entradas, Iterable< Operand < TFloat32 >> learningRates, String config)
Método de fábrica para criar uma classe que agrupa uma nova operação SendTPUEmbeddingGradients.
Parâmetros
| escopo | escopo atual |
|---|---|
| entradas | Um TensorList de gradientes com os quais atualizar tabelas de incorporação. Este argumento tem o mesmo comprimento e formato do valor de retorno de RecvTPUEmbeddingActivations, mas contém gradientes de perda do modelo em relação às ativações de incorporação. As tabelas de incorporação são atualizadas a partir desses gradientes por meio do otimizador especificado na configuração de incorporação da TPU fornecida a tpu.initialize_system. |
| taxas de aprendizagem | Um TensorList de escalares float32, um para cada tag de taxa de aprendizagem dinâmica: veja os comentários em //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Várias tabelas podem compartilhar a mesma tag de taxa de aprendizagem dinâmica conforme especificado na configuração. Se as taxas de aprendizagem para todas as tabelas forem constantes, esta lista deverá estar vazia. |
| configuração | Protocolo TPUEmbeddingConfiguration serializado. |
Devoluções
- uma nova instância de SendTPUEmbeddingGradients