EnqueueTPUEmbeddingBatch

EnqueueTPUEmbeddingBatch kelas akhir publik

Sebuah operasi yang memasukkan daftar tensor batch masukan ke TPUEmbedding.

Sebuah operasi yang memasukkan daftar tensor batch masukan ke TPUEmbedding.

Kelas Bersarang

kelas EnqueueTPUEmbeddingBatch.Options Atribut opsional untuk EnqueueTPUEmbeddingBatch

Metode Publik

EnqueueTPUEmbeddingBatch.Options statis
penggabung (Daftar<String> penggabung)
EnqueueTPUEmbeddingBatch statis
buat ( Lingkup lingkup, Iterable< Operand <String>> batch, Operand <String> modeOverride, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingBatch baru.
EnqueueTPUEmbeddingBatch.Options statis
deviceOrdinal (Perangkat panjangOrdinal)

Metode Warisan

Metode Publik

penggabung EnqueueTPUEmbeddingBatch.Options statis publik (penggabung Daftar<String>)

Parameter
penggabung Daftar skalar string, satu untuk setiap tabel penyematan yang menentukan cara menormalkan aktivasi penyematan setelah penjumlahan berbobot. Penggabung yang didukung adalah 'mean', 'sum', atau 'sqrtn'. Tidak valid jika jumlah bobot menjadi 0 untuk 'rata-rata' atau jumlah bobot kuadrat menjadi 0 untuk 'kuadrat'. Jika penggabung tidak diteruskan, defaultnya adalah menggunakan 'jumlah' untuk semua tabel.

public static EnqueueTPUEmbeddingBatch buat ( Lingkup cakupan , Iterable< Operand <String>> batch, Operand <String> modeOverride, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingBatch baru.

Parameter
cakupan ruang lingkup saat ini
kelompok Daftar tensor 1D, satu untuk setiap tabel penyematan, berisi input batch yang dikodekan sebagai proto dist_belief.SparseFeatures. Jika bidang bobot di proto SparseFeatures tidak diisi untuk ID, bobot diasumsikan 1,0.
modeOverride Input string yang menggantikan mode yang ditentukan dalam TPUEmbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Jika disetel ke 'tidak ditentukan', mode yang disetel di TPUEmbeddingConfiguration akan digunakan, jika tidak, mode_override akan digunakan.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari EnqueueTPUEmbeddingBatch

EnqueueTPUEmbeddingBatch.Options statis publik deviceOrdinal (PerangkatOrdinal panjang)

Parameter
perangkatOrdinal Perangkat TPU yang akan digunakan. Nilainya harus -1 saat Operasi dijalankan pada perangkat TPU, dan >= 0 saat Operasi dijalankan pada perangkat CPU.