Aktivasi Penyematan kelas akhir publik
Sebuah operasi yang memungkinkan diferensiasi Embeddings TPU.
Operasi ini hanya mengembalikan input pertamanya, yang diasumsikan telah dipotong dari Tensor yang dikembalikan oleh TPUEmbeddingDequeueActivations. Kehadiran operasi ini, dan argumen pertamanya adalah Variabel yang dapat dilatih, memungkinkan diferensiasi otomatis grafik yang berisi penyematan melalui pustaka TPU Embedding Python.
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| Keluaran < TFloat32 > | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
| Aktivasi Penyematan statis | |
| Keluaran < TFloat32 > | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "TPUEmbeddingActivations"
Metode Publik
Keluaran publik < TFloat32 > asOutput ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
pembuatan EmbeddingActivations statis publik ( Lingkup cakupan , Operan < TFloat32 > variabel embedding, Operan < TFloat32 > irisanActivations, Long tableId, Long lookupId)
Metode pabrik untuk membuat kelas yang membungkus operasi EmbeddingActivations baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| menyematkanVariabel | Variabel yang dapat dilatih, memungkinkan pengoptimal menemukan operasi ini. |
| irisanAktivasi | Aktivasi penyematan Tensor untuk kembali. |
| tableId | Id tabel dalam konfigurasi lapisan penyematan tempat aktivasi ini dihitung. |
| pencarianId | Pengidentifikasi kumpulan indeks penyematan yang menghasilkan aktivasi ini. |
Kembali
- contoh baru dari EmbeddingActivations