Memuat `Tensor` 2-D (matriks) dengan nama `nama_tensor_lama` dari pos pemeriksaan
di `ckpt_path` dan berpotensi menyusun ulang baris dan kolomnya menggunakan pemetaan ulang yang ditentukan.
Sebagian besar pengguna harus menggunakan salah satu inisialisasi wrapper (seperti `tf.contrib.framework.load_and_remap_matrix_initializer`) daripada menggunakan fungsi ini secara langsung.
Pemetaan ulang adalah tensor 1-D dengan properti berikut:
- `row_remapping` harus memiliki entri `num_rows` yang tepat. Baris `i` dari matriks keluaran akan diinisialisasi dari baris yang sesuai dengan indeks `row_remapping[i]` di `Tensor` lama dari pos pemeriksaan.
- `col_remapping` harus memiliki 0 entri (menunjukkan bahwa tidak diperlukan penyusunan ulang kolom) atau entri `num_cols`. Jika ditentukan, kolom `j` dari matriks keluaran akan diinisialisasi dari kolom yang sesuai dengan indeks `col_remapping[j]` di `Tensor` lama dari pos pemeriksaan.
- Nilai -1 pada salah satu pemetaan ulang menandakan entri "hilang". Dalam hal ini, nilai dari tensor `initializing_values` akan digunakan untuk mengisi baris atau kolom yang hilang tersebut. Jika `row_remapping` memiliki `r` entri yang hilang dan `col_remapping` memiliki `c` entri yang hilang, maka kondisi berikut harus benar:
Tensor pemetaan ulang dapat dibuat menggunakan operasi GenerateVocabRemapping.
Sebagai contoh, dengan row_remapping = [1, 0, -1], col_remapping = [0, 2, -1], initializing_values = [0.5, -0.5, 0.25, -0.25, 42], dan w(i, j) mewakili nilai dari baris i, kolom j tensor lama di checkpoint, matriks keluaran akan terlihat seperti berikut:
[[w(1, 0), w(1, 2), 0,5], [w(0, 0), w(0, 2), -0,5], [0,25, -0,25, 42]]
Kelas Bersarang
kelas | LoadAndRemapMatrix.Options | Atribut opsional untuk LoadAndRemapMatrix |
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran < TFloat32 > | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
LoadAndRemapMatrix statis | buat ( Cakupan cakupan , Operan < TString > ckptPath, Operan < TString > oldTensorName, Operan < TInt64 > rowRemapping, Operan < TInt64 > colRemapping, Operan < TFloat32 > inisialisasiValues, Long numRows, Long numCols, Opsi... opsi) Metode pabrik untuk membuat kelas yang membungkus operasi LoadAndRemapMatrix baru. |
LoadAndRemapMatrix.Options statis | maxRowsInMemory (MaxRowsInMemory panjang) |
Keluaran < TFloat32 > | matriks keluaran () Matriks keluaran berisi nilai-nilai yang ada yang dimuat dari pos pemeriksaan, dan dengan nilai-nilai yang hilang diisi dari nilai_inisialisasi. |
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.
LoadAndRemapMatrix publik statis membuat ( Lingkup lingkup, Operan < TString > ckptPath, Operan < TString > oldTensorName, Operan < TInt64 > rowRemapping, Operan < TInt64 > colRemapping, Operan < TFloat32 > inisialisasiNilai, Long numRows, Long numCols, Pilihan... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi LoadAndRemapMatrix baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
ckptPath | Jalur ke pos pemeriksaan TensorFlow (versi 2, `TensorBundle`) tempat matriks `Tensor` lama akan dimuat. |
namaTensor lama | Nama `Tensor` 2-D yang akan dimuat dari pos pemeriksaan. |
pemetaan ulang baris | Sebuah int `Tensor` dari pemetaan ulang baris (umumnya dibuat oleh `generate_vocab_remapping`). Meskipun pemetaan ulang baris tidak diperlukan, ini tetap harus berupa Tensor dengan nilai indeks (misalnya [0, 1, 2, ...]), atau `Tensor` dengan nilai indeks yang digeser (misalnya [8, 9, 10, ...], untuk `Variabel` yang dipartisi). |
colPemetaan ulang | Sebuah int `Tensor` dari pemetaan ulang kolom (umumnya dibuat oleh `generate_vocab_remapping`). Mungkin `Tensor` ukuran-0 jika hanya pemetaan ulang baris yang dilakukan (misalnya pengurutan kolom sama). |
menginisialisasiNilai | `Tensor` float berisi nilai yang harus diisi untuk sel dalam matriks keluaran yang tidak dimuat dari pos pemeriksaan. Panjangnya harus sama persis dengan jumlah sel yang hilang/baru. |
nomor baris | Jumlah baris (panjang dimensi ke-1) dalam matriks keluaran. |
nomorKolom | Jumlah kolom (panjang dimensi ke-2) pada matriks keluaran. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru LoadAndRemapMatrix
LoadAndRemapMatrix.Options maxRowsInMemory publik statis (MaxRowsInMemory panjang)
Parameter
maxRowsInMemory | Jumlah baris maksimum yang akan dimuat dari pos pemeriksaan sekaligus. Jika kurang dari atau sama dengan 0, seluruh matriks akan dimuat ke dalam memori. Menyetel argumen ini akan meningkatkan pembacaan disk untuk penggunaan memori yang lebih rendah. |
---|