SendTPUEmbeddingGradients

kelas akhir publik SendTPUEmbeddingGradients

Melakukan pembaruan gradien pada tabel yang disematkan.

Konstanta

Rangkaian OP_NAME Nama operasi ini dikenal dengan mesin inti TensorFlow

Metode Publik

SendTPUEmbeddingGradien statis
buat ( Lingkup cakupan , Iterable< Operand < TFloat32 >> input, Iterable< Operand < TFloat32 >> learningRates, String config)
Metode pabrik untuk membuat kelas yang membungkus operasi SendTPUEmbeddingGradients baru.

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "SendTPUEmbeddingGradients"

Metode Publik

public static SendTPUEmbeddingGradients membuat ( Lingkup cakupan, Iterable< Operand < TFloat32 >> input, Iterable< Operand < TFloat32 >> learningRates, String config)

Metode pabrik untuk membuat kelas yang membungkus operasi SendTPUEmbeddingGradients baru.

Parameter
cakupan ruang lingkup saat ini
masukan TensorList gradien yang dapat digunakan untuk memperbarui tabel penyematan. Argumen ini memiliki panjang dan bentuk yang sama dengan nilai kembalian RecvTPUEmbeddingActivations, namun berisi gradien kerugian model sehubungan dengan aktivasi penyematan. Tabel penyematan diperbarui dari gradien ini melalui pengoptimal yang ditentukan dalam konfigurasi penyematan TPU yang diberikan ke tpu.initialize_system.
tarif pembelajaran TensorList skalar float32, satu untuk setiap tag kecepatan pembelajaran dinamis: lihat komentar di //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Beberapa tabel dapat berbagi tag kecepatan pemelajaran dinamis yang sama seperti yang ditentukan dalam konfigurasi. Jika kecepatan pembelajaran untuk semua tabel adalah konstan, daftar ini harus kosong.
konfigurasi Proto TPUEmbeddingConfiguration berseri.
Kembali
  • contoh baru dari SendTPUEmbeddingGradients