CudnnRNNParamsToCanonicalV2

kelas akhir publik CudnnRNNParamsToCanonicalV2

Mengambil parameter CudnnRNN dalam bentuk kanonik. Ini mendukung proyeksi di LSTM.

Mengambil sekumpulan bobot dari buffer params buram yang dapat disimpan dan dipulihkan dengan cara yang kompatibel dengan proses selanjutnya.

Perhatikan bahwa buffer params mungkin tidak kompatibel di berbagai GPU. Jadi setiap penyelamatan dan pemulihan harus dikonversi ke dan dari bobot dan bias kanonik.

num_layers: Menentukan jumlah lapisan dalam model RNN. num_units: Menentukan ukuran status 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 penyimpanan dan pemulihan. Mereka lebih cenderung kompatibel antar generasi yang berbeda. bias: bentuk bias kanonik yang dapat digunakan untuk penyelamatan dan restorasi. Mereka lebih cenderung kompatibel antar generasi yang berbeda. rnn_mode: Menunjukkan jenis model RNN. input_mode: Tunjukkan apakah ada proyeksi linier antara input dan komputasi aktual sebelum lapisan pertama. 'skip_input' hanya diperbolehkan bila input_size == num_units; 'auto_select' menyiratkan 'skip_input' ketika input_size == num_units; jika tidak, ini berarti 'input_linier'. arah: Menunjukkan apakah model dua arah akan digunakan. dir = (arah == dua arah) ? 2: 1 putus sekolah: kemungkinan putus sekolah. Jika disetel ke 0, dropout dinonaktifkan. seed: bagian pertama dari seed untuk menginisialisasi dropout. seed2: bagian ke-2 dari benih untuk menginisialisasi dropout. num_proj: Dimensi keluaran untuk matriks proyeksi. Jika Tidak Ada atau 0, tidak ada proyeksi yang dilakukan.

Kelas Bersarang

kelas CudnnRNNNParamsToCanonicalV2.Options Atribut opsional untuk CudnnRNNParamsToCanonicalV2

Metode Publik

Daftar< Keluaran <T>>
bias ()
statis <T memperluas Nomor> CudnnRNNParamsToCanonicalV2 <T>
buat ( Lingkup lingkup, Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)
Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNNParamsToCanonicalV2 baru.
CudnnRNNParamsToCanonicalV2.Options statis
arah (arah string)
CudnnRNNParamsToCanonicalV2.Options statis
putus sekolah (putus sekolah mengambang)
CudnnRNNParamsToCanonicalV2.Options statis
mode masukan (Mode masukan string)
CudnnRNNParamsToCanonicalV2.Options statis
numProj (NomorProj panjang)
CudnnRNNParamsToCanonicalV2.Options statis
rnnMode (String rnnMode)
CudnnRNNParamsToCanonicalV2.Options statis
benih (biji panjang)
CudnnRNNParamsToCanonicalV2.Options statis
seed2 (Benih panjang2)
Daftar< Keluaran <T>>
beban ()

Metode Warisan

Metode Publik

Daftar publik< Keluaran <T>> bias ()

public static CudnnRNNParamsToCanonicalV2 <T> buat ( Lingkup cakupan , Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)

Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRNNNParamsToCanonicalV2 baru.

Parameter
cakupan ruang lingkup saat ini
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari CudnnRNNNParamsToCanonicalV2

public static CudnnRNNParamsToCanonicalV2.Arah opsi (Arah string)

CudnnRNNParamsToCanonicalV2.Options statis publik putus sekolah (Float dropout)

CudnnRNNParamsToCanonicalV2.Options inputMode statis publik (String inputMode)

CudnnRNNParamsToCanonicalV2.Options numProj statis publik (NumProj panjang)

CudnnRNNParamsToCanonicalV2.Options statis publik rnnMode (String rnnMode)

public static CudnnRNNParamsToCanonicalV2.Options seed (Benih panjang)

public static CudnnRNNParamsToCanonicalV2.Options seed2 (Benih panjang2)

Daftar publik< Output <T>> bobot ()