SendTPUEmbeddingGradients

SendTPUEmbeddingGradients classe final públicas

Executa atualizações de gradiente de tabelas de incorporação.

Métodos Públicos

estáticos SendTPUEmbeddingGradients
criar ( Scope escopo, Iterable < Operando <Float >> entradas, Iterable < Operando <Float >> learningRates, String config)
Método de fábrica para criar uma classe que envolve uma nova operação SendTPUEmbeddingGradients.

Métodos herdados

Métodos Públicos

public static SendTPUEmbeddingGradients criar ( Scope escopo, Iterable < Operando <Float >> entradas, Iterable < Operando <Float >> 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 aprendizagem para todas as tabelas forem constantes, esta lista deve estar vazia.
config Protocolo TPUEmbeddingConfiguration serializado.
Devoluções
  • uma nova instância de SendTPUEmbeddingGradients