kelas akhir publik EnqueueTPUEmbeddingBatch
Sebuah op yang mengantrekan daftar tensor batch input ke TPUEbedding.
Sebuah op yang mengantrekan daftar tensor batch input ke TPUEbedding.
Kelas Bersarang
kelas | EnqueueTPUEbeddingBatch.Options | Atribut opsional untuk EnqueueTPUEmbeddingBatch |
Metode Publik
EnqueueTPUEmbeddingBatch.Options statis | penggabung (penggabung Daftar<String>) |
EnqueueTPUembeddingBatch statis | |
EnqueueTPUEmbeddingBatch.Options statis | deviceOrdinal (Long deviceOrdinal) |
Metode yang Diwarisi
Metode Publik
penggabung EnqueueTPUEmbeddingBatch.Options publik statis ( 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 'persegi'. Jika penggabung tidak lulus, defaultnya adalah menggunakan 'jumlah' untuk semua tabel. |
---|
public static EnqueueTPUEmbeddingBatch create ( Scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingBatch baru.
Parameter
cakupan | 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 1,0 diasumsikan. |
modeOverride | Input string yang menggantikan mode yang ditentukan dalam TPUEmbbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Ketika diatur ke 'tidak ditentukan', mode yang diatur dalam TPUEmbeddingConfiguration digunakan, jika tidak, mode_override digunakan. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari EnqueueTPUEmbeddingBatch
EnqueueTPUEmbeddingBatch.Options publik statis deviceOrdinal ( Long deviceOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Ini harus -1 ketika Op berjalan pada perangkat TPU, dan >= 0 saat Op berjalan pada perangkat CPU. |
---|