CudnnRnnParamsSize

kelas akhir publik CudnnRnnParamsSize

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>
buat ( Lingkup lingkup, 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.
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>
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

Nilai Konstan: "CudnnRNNParamsSize"

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

arah CudnnRnnParamsSize.Options statis publik (arah string)

putus sekolah CudnnRnnParamsSize.Options statis publik (putus sekolah Float)

CudnnRnnParamsSize.Options statis publik inputMode (String inputMode)

CudnnRnnParamsSize.Options numProj statis publik (NumProj panjang)

Keluaran publik <U> paramsSize ()

CudnnRnnParamsSize.Options statis publik rnnMode (String rnnMode)

benih CudnnRnnParamsSize.Options statis publik (Benih panjang)

public static CudnnRnnParamsSize.Options seed2 (Benih panjang2)