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
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