Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023 Registre-se

Esta página foi traduzida pela API Cloud Translation.

SendTPUEmbeddingGradients

classe final pública SendTPUEmbeddingGradients

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

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.

booleano final	igual a (objeto obj)
final int	código de hash ()
Operação	op () Retorna a `Operation` subjacente
String final	toString ()

Da classe java.lang.Object

Métodos públicos

Método de fábrica para criar uma classe envolvendo uma nova operação SendTPUEmbeddingGradients.

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.