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