XlaSendTPUEmbeddingGradients

kelas akhir publik XlaSendTPUEmbeddingGradients

Operasi yang melakukan pembaruan gradien tabel penyematan.

Argumen gradien adalah TensorList yang memiliki panjang dan bentuk yang sama dengan nilai kembalian XlaRecvTPUEmbeddingActivations, tetapi berisi gradien hilangnya model sehubungan dengan aktivasi penyematan. Tabel penyematan diperbarui dari gradien ini melalui pengoptimal yang ditentukan dalam proto TPUEmbeddingConfiguration yang diberikan ke tpu.initialize_system.

Metode Publik

XlaSendTPUEmbeddingGradients statis

create ( Scope scope, Iterable< Operand <Float>> gradien, Iterable< Operand <Float>> learningRates, Operand <?> deduplicationData, String config) Metode pabrik untuk membuat kelas yang membungkus operasi XlaSendTPUEmbeddingGradients baru.

Metode yang Diwarisi

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (objek objek)
akhir	kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkanKelas ()
ke dalam	kode hash ()
kekosongan akhir	beri tahu ()
kekosongan akhir	beri tahu Semua ()
Rangkaian	toString ()
kekosongan akhir	tunggu (arg panjang0, int arg1)
kekosongan akhir	tunggu (arg panjang0)
kekosongan akhir	tunggu ()

Metode Publik

public static XlaSendTPUEmbeddingGradients create ( Scope scope, Iterable< Operand <Float>> gradients, Iterable< Operand <Float>> learningRates, Operand <?> deduplicationData, String config)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaSendTPUEmbeddingGradients baru.

Parameter

cakupan	lingkup saat ini
gradien	TensorList gradien yang digunakan untuk memperbarui tabel penyematan.
Tarif belajar	TensorList kecepatan pembelajaran yang digunakan untuk memperbarui tabel penyematan melalui pengoptimal. Panjang TensorList harus sama dengan jumlah tag kecepatan pembelajaran dinamis yang ditentukan dalam proto TPUEmbeddingConfiguration.
data deduplikasi	Tensor dengan type=DT_VARIANT yang berisi data deduplikasi. Tensor adalah tupel bersarang XLA yang berisi elemen N (dengan N adalah rasio jumlah penyematan ke inti tensor per chip TPU). Setiap elemen tupel bersarang adalah tupel tensor peringkat 1. Setiap tensor berisi indeks (DT_UINT32) untuk menyematkan pencarian pada TensorCore atau bobot (DT_FLOAT) untuk diterapkan ke output dari operasi pencarian penyematan.
konfigurasi	Proto konfigurasi TPUEbeddingConfiguration berseri.

Kembali

• contoh baru dari XlaSendTPUEmbeddingGradients