kelas akhir publik EnqueueTPUEmbeddingIntegerBatch
Sebuah operasi yang memasukkan daftar tensor batch masukan ke TPUEmbedding.
Kelas Bersarang
kelas | EnqueueTPUEmbeddingIntegerBatch.Options | Atribut opsional untuk EnqueueTPUEmbeddingIntegerBatch |
Metode Publik
EnqueueTPUEmbeddingIntegerBatch statis | |
EnqueueTPUEmbeddingIntegerBatch.Options statis | deviceOrdinal (Perangkat panjangOrdinal) |
Metode Warisan
Metode Publik
public static EnqueueTPUEmbeddingIntegerBatch buat ( Lingkup cakupan , Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
kelompok | Daftar tensor 1D, satu untuk setiap tabel penyematan, berisi indeks ke dalam tabel. |
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 EnqueueTPUEmbeddingIntegerBatch
Enqueuestatis publikTPUEmbeddingIntegerBatch.Options deviceOrdinal (PerangkatOrdinal panjang)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas tempat node ditempatkan. |
---|