SendTPUEmbeddingGradients

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
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.

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