Mengambil params CudnnRNN dalam bentuk kanonik. Ini mendukung proyeksi di LSTM.
Mengambil satu set bobot dari buffer params buram yang dapat disimpan dan dipulihkan dengan cara yang kompatibel dengan proses di masa mendatang.
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. num_params_weights: jumlah matriks parameter bobot untuk semua lapisan. num_params_biases: jumlah vektor parameter bias untuk semua lapisan. 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. 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 | CudnnRNNParamsToCanonicalV2.Options | Atribut opsional untuk CudnnRNNParamsToCanonicalV2 |
Metode Publik
Daftar < Keluaran <T >> | bias () |
statis <T meluas Nomor> CudnnRNNParamsToCanonicalV2 <T> | |
statis CudnnRNNParamsToCanonicalV2.Options | arah (arah String) |
statis CudnnRNNParamsToCanonicalV2.Options | putus sekolah (Lampung putus sekolah) |
statis CudnnRNNParamsToCanonicalV2.Options | inputMode (String inputMode) |
statis CudnnRNNParamsToCanonicalV2.Options | numProj (Long numProj) |
statis CudnnRNNParamsToCanonicalV2.Options | rnnMode (String rnnMode) |
statis CudnnRNNParamsToCanonicalV2.Options | benih (Long biji) |
statis CudnnRNNParamsToCanonicalV2.Options | seed2 (Long seed2) |
Daftar < Keluaran <T >> | bobot () |
Metode yang Diwarisi
Metode Publik
public static CudnnRNNParamsToCanonicalV2 <T> membuat ( Lingkup lingkup, Operan <Integer> numLayers, Operan <Integer> numUnits, Operan <Integer> inputSize, Operan <T> params, Long numParamsWeights, Long numParamsBiases, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNParamsToCanonicalV2 baru.
Parameter
cakupan | lingkup saat ini |
---|---|
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari CudnnRNNParamsToCanonicalV2