RetrieveAllTPUEmbeddingParameters

kelas akhir publik RetrieveAllTPUEmbeddingParameters

Sebuah operasi yang mengambil parameter optimasi dari penyematan ke memori host.

Sebuah operasi yang mengambil parameter optimasi dari penyematan ke memori host. Harus didahului dengan operasi ConfigureTPUEmbeddingHost 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 tersebut). Untuk FTRL, auxiliary1 akan berisi akumulator dan auxiliary2 akan berisi suku-suku linier. Untuk ADAM, auxiliary1 berisi momentum dan auxiliary2 berisi kecepatan.

Metode Publik

Daftar< Keluaran <Mengambang>>
bantu1 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan pertama yang disimpan.
Daftar< Keluaran <Mengambang>>
bantu2 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan kedua yang disimpan.
Daftar< Keluaran <Mengambang>>
bantu3 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketiga yang disimpan.
Daftar< Keluaran <Mengambang>>
tambahan4 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan keempat yang disimpan.
Daftar< Keluaran <Mengambang>>
bantu5 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan kelima yang disimpan.
Daftar< Keluaran <Mengambang>>
tambahan6 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi enam parameter pengoptimalan tambahan yang disimpan.
Daftar< Keluaran <Mengambang>>
tambahan7 ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketujuh yang disimpan.
RetrieveAllTPUEmbeddingParameters statis
buat ( Lingkup cakupan, Long NumTables, Konfigurasi string, Long numShards, Long shardId)
Metode pabrik untuk membuat kelas yang membungkus operasi RetrieveAllTPUEmbeddingParameters baru.
Daftar< Keluaran <Mengambang>>
parameter ()
Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter tabel penyematan yang disimpan.

Metode Warisan

Metode Publik

Daftar publik< Keluaran <Float>> tambahan1 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan pertama yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan2 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan kedua yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan3 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketiga yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan4 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan keempat yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan5 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan kelima yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan6 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi enam parameter pengoptimalan tambahan yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

Daftar publik< Keluaran <Float>> tambahan7 ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter pengoptimalan tambahan ketujuh yang disimpan. Elemen ada dalam daftar, namun memiliki ukuran nol, untuk parameter optimasi yang tidak digunakan (berdasarkan algoritma yang digunakan untuk setiap tabel).

public static RetrieveAllTPUEmbeddingParameters buat ( Lingkup cakupan , Long NumTables, Konfigurasi string, Long numShards, Long shardId)

Metode pabrik untuk membuat kelas yang membungkus operasi RetrieveAllTPUEmbeddingParameters baru.

Parameter
cakupan ruang lingkup saat ini
Nomor Tabel Jumlah tabel yang disematkan.
konfigurasi Proto TPUEmbeddingConfiguration yang menjelaskan parameter tabel yang dimuat, diserialkan ke string.
jumlah pecahan Jumlah pecahan yang menjadi tempat pembagian tabel penyematan.
pecahanId Pengidentifikasi pecahan untuk operasi ini.
Kembali
  • contoh baru dari RetrieveAllTPUEmbeddingParameters

Daftar publik< Output <Float>> parameter ()

Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter tabel penyematan yang disimpan.