Sebuah op yang menerima aktivasi embedding pada TPU.
Sistem TPU melakukan pencarian dan agregasi penyematan yang ditentukan oleh argumen ke TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. Hasil agregasi ini terlihat oleh Grafik Tensorflow sebagai output dari operasi RecvTPUEmbbeddingActivations. Operasi ini mengembalikan daftar yang berisi satu Tensor aktivasi per tabel yang ditentukan dalam model. Paling banyak ada satu operasi RecvTPUEmbeddingActivations dalam grafik TPU.
Metode Publik
Aktivasi RecvTPUEbedding statis | |
Iterator< Operan <Float>> | pengulangan () |
Daftar< Keluaran <Float>> | keluaran () TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model. |
Metode yang Diwarisi
Metode Publik
Buat RecvTPUEmbbeddingActivations statis publik ( Cakupan lingkup, Jumlah panjang, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi RecvTPUEmbbeddingActivations baru.
Parameter
cakupan | lingkup saat ini |
---|---|
jumlah Output | Jumlah tensor aktivasi keluaran, sama dengan jumlah tabel embedding dalam model. |
konfigurasi | Proto konfigurasi TPUEbeddingConfiguration berseri. |
Kembali
- contoh baru dari RecvTPUEmbeddingActivations
Daftar publik< Output <Float>> output ()
TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model.