SendTPUEmbeddingGradients

classe final pública SendTPUEmbeddingGradients

Executa atualizações de gradiente de tabelas incorporadas.

Métodos Públicos

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

Métodos herdados

Métodos Públicos

public static SendTPUEmbeddingGradients create ( escopo de escopo , entradas Iterable< Operand <Float>>, Iterable< Operand <Float>> 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