EnqueueTPUEmbeddingIntegerBatch

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
buat ( Lingkup lingkup, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru.
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.