Langkah backprop dari CudnnRNNV3.
Hitung backprop dari kedua data dan bobot dalam RNN. Mengambil input "sequence_lengths" ekstra daripada CudnnRNNBackprop.
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. Harus "searah" atau "dua arah". 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. input: Jika time_major benar, ini adalah tensor 3-D dengan bentuk [seq_length, batch_size, input_size]. Jika time_major salah, bentuknya adalah [batch_size, seq_length, input_size]. input_h: Jika time_major benar, ini adalah tensor 3-D dengan bentuk [num_layer * dir, batch_size, num_units]. Jika time_major salah, bentuknya adalah [batch_size, num_layer * dir, num_units]. input_c: Untuk LSTM, tensor 3-D dengan bentuk [num_layer * dir, batch, num_units]. Untuk model lain, diabaikan. params: Tensor 1-D yang berisi bobot dan bias dalam tata letak buram. Ukuran harus dibuat melalui CudnnRNNParamsSize, dan diinisialisasi secara terpisah. Perhatikan bahwa mereka mungkin tidak kompatibel di berbagai generasi. Jadi sebaiknya simpan dan pulihkan sequence_lengths: vektor panjang dari setiap urutan input. output: Jika time_major benar, ini adalah tensor 3-D dengan bentuk [seq_length, batch_size, dir * num_units]. Jika time_major salah, bentuknya adalah [batch_size, seq_length, dir * num_units]. output_h: Bentuk yang sama memiliki input_h. output_c: Bentuk yang sama dengan input_c untuk LSTM. Tensor kosong untuk model lain. output_backprop: Tensor 3-D dengan bentuk yang sama seperti output pada lintasan maju. output_h_backprop: Tensor 3-D dengan bentuk yang sama seperti output_h pada forward pass. output_c_backprop: Tensor 3-D dengan bentuk yang sama dengan output_c pada forward pass. time_major: Menunjukkan apakah format input/output adalah mayor waktu atau mayor batch. reserve_space: Reserve_space yang sama yang dihasilkan dalam operasi forward. input_backprop: Backprop untuk memasukkan umpan maju. Memiliki bentuk yang sama dengan input. input_h_backprop: Backprop ke input_h di forward pass. Memiliki bentuk yang sama dengan input_h. input_c_backprop: Backprop ke input_c di forward pass. Memiliki bentuk yang sama dengan input_c. params_backprop: Backprop ke buffer params di forward pass. Memiliki bentuk yang sama dengan params.
Kelas Bersarang
kelas | CudnnRNNBackpropV3.Options | Atribut opsional untuk CudnnRNNBackpropV3 |
Metode Publik
statis <T meluas Nomor> CudnnRNNBackpropV3 <T> | membuat ( Lingkup lingkup, Operan <T> input, Operan <T> inputH, Operan <T> inputC, Operan <T> params, Operan <Integer> sequenceLengths, Operan <T> output, Operan <T> outputH, Operan <T > outputC, Operan <T> outputBackprop, Operan <T> outputHBackprop, Operan <T> outputCBackprop, Operan <T> reserveSpace, Operan <?> hostReserved, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNBackpropV3 baru. |
statis CudnnRNNBackpropV3.Options | arah (arah String) |
statis CudnnRNNBackpropV3.Options | putus sekolah (Lampung putus sekolah) |
Output <T> | |
Output <T> | |
Output <T> | |
statis CudnnRNNBackpropV3.Options | inputMode (String inputMode) |
statis CudnnRNNBackpropV3.Options | numProj (Long numProj) |
Output <T> | |
statis CudnnRNNBackpropV3.Options | rnnMode (String rnnMode) |
statis CudnnRNNBackpropV3.Options | benih (Long biji) |
statis CudnnRNNBackpropV3.Options | seed2 (Long seed2) |
statis CudnnRNNBackpropV3.Options | timeMajor (Boolean timeMajor) |
Metode yang Diwarisi
Metode Publik
public static CudnnRNNBackpropV3 <T> membuat ( Lingkup lingkup, Operan <T> input, Operan <T> inputH, Operan <T> inputC, Operan <T> params, Operan <Integer> sequenceLengths, Operan <T> output, Operan <T > outputH, Operan <T> outputC, Operan <T> outputBackprop, Operan <T> outputHBackprop, Operan <T> outputCBackprop, Operan <T> reserveSpace, Operan <?> hostReserved, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNBackpropV3 baru.
Parameter
cakupan | lingkup saat ini |
---|---|
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari CudnnRNNBackpropV3