XlaRecvTPUEmbeddingDeduplicationData

kelas akhir publik XlaRecvTPUEmbeddingDeduplicationData

Menerima data deduplikasi (indeks dan bobot) dari inti penyematan.

Data deduplikasinya adalah Tensor dengan type=DT_VARIANT. Tensor itu sendiri adalah tupel bersarang XLA yang berisi N elemen (di mana N adalah rasio jumlah penyematan inti tensor per chip TPU). Setiap elemen tupel bersarang adalah tupel dengan tensor peringkat 1. Setiap tensor berisi indeks (DT_UINT32) untuk menyematkan pencarian di TensorCore atau bobot (DT_FLOAT) untuk diterapkan ke output dari operasi pencarian penyematan.

Metode Publik

Keluaran <Objek>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
XlaRecvTPUEmbeddingDeduplicationData statis
buat ( Lingkup lingkup, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.
Keluaran <?>

Metode Warisan

Metode Publik

Output publik <Objek> asOutput ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

pembuatan XlaRecvTPUEmbeddingDeduplicationData statis publik ( Lingkup cakupan, Konfigurasi string)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.

Parameter
cakupan ruang lingkup saat ini
konfigurasi Proto TPUEmbeddingConfiguration berseri.
Kembali
  • contoh baru dari XlaRecvTPUEmbeddingDeduplicationData

Keluaran publik <?> keluaran ()