Menghitung ukuran bobot yang dapat digunakan oleh model Cudnn RNN.
Mengembalikan ukuran params yang dapat digunakan oleh model Cudnn RNN. Alokasi bobot dan inisialisasi selanjutnya harus menggunakan ukuran ini.
num_layers: Menentukan jumlah lapisan dalam model RNN. num_units: Menentukan ukuran status tersembunyi. input_size: Menentukan ukuran status input. 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. params_size: Ukuran buffer params yang harus dialokasikan dan diinisialisasi untuk model RNN ini. Perhatikan bahwa buffer params ini mungkin tidak kompatibel di seluruh GPU. Silakan gunakan CudnnRNNParamsWeights dan CudnnRNNParamsBiases untuk menyimpan dan memulihkannya dengan cara yang kompatibel di berbagai proses.
Kelas Bersarang
kelas | CudnnRnnParamsSize.Options | Atribut opsional untuk CudnnRnnParamsSize |
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <U> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
statis <U memperluas TNomber , T memperluas TNomber > CudnnRnnParamsSize <U> | |
CudnnRnnParamsSize.Options statis | arah (arah string) |
CudnnRnnParamsSize.Options statis | putus sekolah (putus sekolah mengambang) |
CudnnRnnParamsSize.Options statis | mode masukan (Mode masukan string) |
CudnnRnnParamsSize.Options statis | numProj (NomorProj panjang) |
Keluaran <U> | paramsUkuran () |
CudnnRnnParamsSize.Options statis | rnnMode (String rnnMode) |
CudnnRnnParamsSize.Options statis | benih (biji panjang) |
CudnnRnnParamsSize.Options statis | seed2 (Benih panjang2) |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <U> sebagai Keluaran ()
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.
public static CudnnRnnParamsSize <U> buat ( Lingkup cakupan, Operan < TInt32 > numLayers, Operan < TInt32 > numUnits, Operan < TInt32 > inputSize, Kelas<T> T, Kelas<U> S, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi CudnnRnnParamsSize baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari CudnnRnnParamsSize