SendTPUEmbeddingGradients classe final públicas
Executa atualizações de gradiente de tabelas de incorporação.
Constantes
| Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
| estáticos SendTPUEmbeddingGradients |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Valor constante: "SendTPUEmbeddingGradients"
Métodos Públicos
public static SendTPUEmbeddingGradients criar ( Scope escopo, Iterable < Operando < TFloat32 >> entradas, Iterable < Operando < TFloat32 >> learningRates, String config)
Método de fábrica para criar uma classe que envolve uma nova operação SendTPUEmbeddingGradients.
Parâmetros
| alcance | escopo atual |
|---|---|
| entradas | Um TensorList de gradientes para atualizar as tabelas de incorporação. Este argumento tem o mesmo comprimento e formato que o valor de retorno de RecvTPUEmbeddingActivations, mas contém gradientes da 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. |
| learningRates | Uma 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 etiqueta de taxa de aprendizagem dinâmica, conforme especificado na configuração. Se as taxas de aprendizado para todas as tabelas forem constantes, esta lista deve estar vazia. |
| config | Protocolo TPUEmbeddingConfiguration serializado. |
Devoluções
- uma nova instância de SendTPUEmbeddingGradients