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.
Metode Publik
Keluaran <Mengambang> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
Aktivasi TPUEmbedding statis | |
Keluaran <Mengambang> | keluaran () |
Metode Warisan
Metode Publik
Keluaran publik <Float> sebagaiOutput ()
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.
TPUEmbeddingActivations statis publik dibuat ( Lingkup cakupan, Operan <Float> embeddingVariable, Operan <Float> irisanActivations, Long tableId, Long lookupId)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUEmbeddingActivations 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 TPUEmbeddingActivations