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