public class akhir EnqueueTPUEmbeddingIntegerBatch
Sebuah op yang mengantrekan daftar tensor batch input ke TPUEbedding.
Kelas Bersarang
kelas | EnqueueTPUEbeddingIntegerBatch.Options | Atribut opsional untuk EnqueueTPUEmbeddingIntegerBatch |
Metode Publik
statis EnqueueTPUEmbeddingIntegerBatch | membuat ( Lingkup lingkup, Iterable < Operan <Integer >> batch, Operan <String> modeOverride, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru. |
statis EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
Metode yang Diwarisi
Metode Publik
public static EnqueueTPUEmbeddingIntegerBatch membuat ( Scope lingkup, Iterable < Operan <Integer >> batch, Operan <String> modeOverride, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru.
Parameter
cakupan | lingkup saat ini |
---|---|
kelompok | Daftar tensor 1D, satu untuk setiap tabel penyematan, yang berisi indeks ke dalam tabel. |
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 EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas di mana node ditempatkan. |
---|