Mengonversi params CudnnRNN dari bentuk kanonik ke bentuk yang dapat digunakan. Ini mendukung proyeksi di LSTM.
Menulis satu set bobot ke dalam buffer params buram sehingga dapat digunakan dalam pelatihan atau inferensi yang akan datang.
Perhatikan bahwa buffer params mungkin tidak kompatibel di seluruh GPU yang berbeda. Jadi setiap penyimpanan dan restorasi harus dikonversi ke dan dari bobot dan bias kanonik.
num_layers: Menentukan jumlah lapisan dalam model RNN. num_units: Menentukan ukuran keadaan tersembunyi. input_size: Menentukan ukuran status input. bobot: bentuk bobot kanonik yang dapat digunakan untuk menyimpan dan memulihkan. Mereka lebih cenderung kompatibel di berbagai generasi. bias: bentuk bias kanonik yang dapat digunakan untuk menyimpan dan memulihkan. Mereka lebih cenderung kompatibel di berbagai generasi. num_params_weights: jumlah matriks parameter bobot untuk semua lapisan. num_params_biases: jumlah vektor parameter bias untuk semua lapisan. rnn_mode: Menunjukkan tipe model RNN. input_mode: Menunjukkan apakah ada proyeksi linier antara input dan Perhitungan sebenarnya sebelum lapisan pertama. 'skip_input' hanya diperbolehkan jika input_size == num_units; 'auto_select' menyiratkan 'skip_input' ketika input_size == num_units; jika tidak, ini menyiratkan 'linear_input'. arah: Menunjukkan apakah model dua arah akan digunakan. dir = (arah == dua arah) ? 2 : 1 putus sekolah: probabilitas putus sekolah. Jika disetel ke 0, dropout dinonaktifkan. seed: bagian pertama dari seed untuk menginisialisasi putus sekolah. seed2: bagian ke-2 dari seed untuk menginisialisasi putus sekolah. num_proj: Dimensi keluaran untuk matriks proyeksi. Jika Tidak Ada atau 0, tidak ada proyeksi yang dilakukan.
Kelas Bersarang
| kelas | CudnnRNNCanonicalToParamsV2.Options | Atribut opsional untuk CudnnRNNCanonicalToParamsV2 | |
Metode Publik
| Output <T> | asOutput () Mengembalikan pegangan simbolis tensor. |
| statis <T meluas Nomor> CudnnRNNCanonicalToParamsV2 <T> | |
| statis CudnnRNNCanonicalToParamsV2.Options | arah (arah String) |
| statis CudnnRNNCanonicalToParamsV2.Options | putus sekolah (Lampung putus sekolah) |
| statis CudnnRNNCanonicalToParamsV2.Options | inputMode (String inputMode) |
| statis CudnnRNNCanonicalToParamsV2.Options | numProj (Long numProj) |
| Output <T> | params () |
| statis CudnnRNNCanonicalToParamsV2.Options | rnnMode (String rnnMode) |
| statis CudnnRNNCanonicalToParamsV2.Options | benih (Long biji) |
| statis CudnnRNNCanonicalToParamsV2.Options | seed2 (Long seed2) |
Metode yang Diwarisi
Metode Publik
publik Keluaran <T> asOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static CudnnRNNCanonicalToParamsV2 <T> membuat ( Lingkup lingkup, Operan <Integer> numLayers, Operan <Integer> numUnits, Operan <Integer> inputSize, Iterable < Operan <T >> bobot, Iterable < Operan <T >> bias, Options .. . Pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNCanonicalToParamsV2 baru.
Parameter
| cakupan | lingkup saat ini |
|---|---|
| pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari CudnnRNNCanonicalToParamsV2