Sebuah operasi yang menerima aktivasi penyematan pada TPU.
Sistem TPU melakukan pencarian dan agregasi penyematan. Hasil agregasi ini dapat dilihat oleh Grafik Tensorflow sebagai output dari Operasi XlaRecvTPUEmbeddingActivations. Operasi ini mengembalikan daftar yang berisi satu Tensor aktivasi per tabel yang ditentukan dalam model.
Metode Publik
Aktivasi XlaRecvTPUEmbedding statis | |
Iterator< Operan <Float>> | pengulangan () |
Daftar< Keluaran <Mengambang>> | keluaran () TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model. |
Metode Warisan
Metode Publik
pembuatan XlaRecvTPUEmbeddingActivations statis publik ( Lingkup cakupan, Operan <?> deduplicationData, Long numTables, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingActivations baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
deduplikasiData | Tensor dengan type=DT_VARIANT berisi data deduplikasi. Tensor adalah tupel bersarang XLA yang berisi N elemen (dengan 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. |
nomor Tabel | Jumlah tensor aktivasi keluaran. Jika deskriptor fitur ada dalam konfigurasi penyematan TPU, itu sama dengan jumlah fitur, jika tidak, sama dengan jumlah tabel penyematan dalam model. |
konfigurasi | Proto TPUEmbeddingConfiguration berseri. |
Kembali
- contoh baru dari XlaRecvTPUEmbeddingActivations
Daftar publik< Keluaran <Float>> keluaran ()
TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model.