classe final pública SendTPUEmbeddingGradients
Executa atualizações de gradiente de tabelas de incorporação.
Métodos públicos
static SendTPUEmbeddingGradients |
Métodos Herdados
Métodos públicos
public static SendTPUEmbeddingGradients create ( Escopo do escopo, Iterable< Operando <Float>> entradas, Iterable< Operando <Float>> learningRates, String config)
Método de fábrica para criar uma classe envolvendo 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 formas que o 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 | Uma TensorList de escalares float32, um para cada tag de taxa de aprendizado dinâmica: consulte os comentários em //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Várias tabelas podem compartilhar a mesma tag de taxa de aprendizado dinâmica conforme especificado na configuração. Se as taxas de aprendizado de todas as tabelas forem constantes, essa lista deverá estar vazia. |
configuração | Protocolo serializado de configuração TPUEmbedding. |
Devoluções
- uma nova instância de SendTPUEmbeddingGradients