TPUEmbeddingActivations

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

Metode Publik

Keluaran <Mengambang>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Aktivasi TPUEmbedding statis
buat ( Lingkup lingkup, Operan <Float> embeddingVariable, Operan <Float> irisanActivations, Long tableId, Long lookupId)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUEmbeddingActivations baru.
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

Keluaran publik <Mengambang> keluaran ()