Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

XlaRecvTPUEmbeddingDeduplicationData

kelas akhir publik XlaRecvTPUEmbeddingDeduplicationData

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

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

Metode Publik

Keluaran <Objek>
sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
XlaRecvTPUEmbeddingDeduplicationData statis
buat ( Cakupan lingkup, konfigurasi String)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.
Keluaran <?>

Metode yang Diwarisi

Metode Publik

Output publik <Object> sebagai Output ()

Mengembalikan pegangan simbolis tensor.

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

Buat XlaRecvTPUEmbeddingDeduplicationData statis publik ( Cakupan lingkup, Konfigurasi string)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.

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

Keluaran publik <?> keluaran ()