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 TPUEmbedding 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.
TPUEmbeddingActivations statis publik dibuat ( Lingkup cakupan, Operan < TFloat32 > variabel embedding, Operan < TFloat32 > irisanActivations, TableId panjang, LookupId panjang)
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