Operasi yang mengambil parameter pengoptimalan dari penyematan ke memori host.
Operasi yang mengambil parameter pengoptimalan dari penyematan ke memori host. Harus didahului oleh operasi ConfigureTPUEmbbeddingHost yang menyiapkan konfigurasi tabel penyematan yang benar. Misalnya, operasi ini digunakan untuk mengambil parameter yang diperbarui sebelum menyimpan pos pemeriksaan. Untuk Adagrad, auxiliary1 akan berisi akumulator setelah menjalankan operasi ini. Untuk SGD, semua nilai tambahan* akan kosong (tensor (0x0 untuk tabel itu). Untuk FTRL, bantu1 akan berisi akumulator dan bantu2 akan berisi istilah linier. Untuk ADAM, bantu1 akan berisi momentum dan bantu2 akan berisi kecepatan.
Metode Publik
Daftar< Keluaran <Float>> | bantu1 () Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan bantu pertama yang disimpan. |
Daftar< Keluaran <Float>> | bantu2 () Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan kedua yang disimpan. |
Daftar< Keluaran <Float>> | bantu3 () Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketiga yang disimpan. |
Daftar< Keluaran <Float>> | bantu4 () Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan keempat yang disimpan. |
Daftar< Keluaran <Float>> | bantu5 () Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan bantu kelima yang disimpan. |
Daftar< Keluaran <Float>> | bantu6 () Daftar tensor, satu untuk setiap tabel penyematan, berisi enam parameter pengoptimalan tambahan yang disimpan. |
Daftar< Keluaran <Float>> | bantu7 () Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan ketujuh yang disimpan. |
static RetrieveAllTPUEmbeddingParameters | |
Daftar< Keluaran <Float>> | parameter () Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter tabel penyematan yang disimpan. |
Metode yang Diwarisi
Metode Publik
Daftar publik< Output <Float>> tambahan1 ()
Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan bantu pertama yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan2 ()
Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan kedua yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan3 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketiga yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan4 ()
Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan keempat yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan5 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan bantu kelima yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan6 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi enam parameter pengoptimalan tambahan yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
Daftar publik< Output <Float>> tambahan7 ()
Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter pengoptimalan tambahan ketujuh yang disimpan. Elemen ada dalam daftar, tetapi memiliki ukuran nol, untuk parameter pengoptimalan yang tidak digunakan (berdasarkan algoritme yang digunakan untuk setiap tabel).
public static RetrieveAllTPUEmbeddingParameters create ( Lingkup lingkup, Long NumTables, String config, Long numShards, Long shardId)
Metode pabrik untuk membuat kelas yang membungkus operasi RetrieveAllTPUEmbeddingParameters baru.
Parameter
cakupan | lingkup saat ini |
---|---|
Tabel Angka | Jumlah tabel penyematan. |
konfigurasi | Proto TPUEmbeddingConfiguration yang menjelaskan parameter tabel yang dimuat, diserialisasi ke string. |
numShards | Jumlah pecahan tempat tabel penyematan dibagi. |
shardId | Pengenal pecahan untuk operasi ini. |
Kembali
- contoh baru dari RetrieveAllTPUEmbeddingParameters
Daftar publik< Output <Float>> parameter ()
Daftar tensor, satu untuk setiap tabel penyematan, yang berisi parameter tabel penyematan yang disimpan.