Cudnn RNN modeli tarafından kullanılabilecek ağırlıkların boyutunu hesaplar.
Cudnn RNN modeli tarafından kullanılabilecek parametre boyutunu döndürün. Sonraki ağırlık tahsisi ve başlatma bu boyutu kullanmalıdır.
num_layers: RNN modelindeki katman sayısını belirtir. num_units: Gizli durumun boyutunu belirtir. input_size: Giriş durumunun boyutunu belirtir. rnn_mode: RNN modelinin tipini belirtir. input_mode: Giriş ile ilk katmandan önceki gerçek hesaplama arasında doğrusal bir projeksiyon olup olmadığını belirtin. 'skip_input'a yalnızca input_size == num_units; olduğunda izin verilir. 'auto_select', input_size == num_units; olduğunda 'skip_input' anlamına gelir; aksi takdirde 'doğrusal_giriş' anlamına gelir. Direction: Çift yönlü bir modelin kullanılıp kullanılmayacağını belirtir. yön = (yön == çift yönlü) ? 2 : 1 bırakma: bırakma olasılığı. 0'a ayarlandığında bırakma devre dışı bırakılır. tohum: bırakmayı başlatan tohumun 1. kısmı. tohum2: bırakmayı başlatmak için bir tohumun 2. kısmı. params_size: Bu RNN modeli için ayrılması ve başlatılması gereken params arabelleğinin boyutu. Bu parametre arabelleğinin GPU'lar arasında uyumlu olmayabileceğini unutmayın. Lütfen bunları farklı çalıştırmalarla uyumlu bir şekilde kaydetmek ve geri yüklemek için CudnnRNNParamsWeights ve CudnnRNNParamsBiases'i kullanın.
İç İçe Sınıflar
sınıf | CudnnRnnParamsSize.Options | CudnnRnnParamsSize için isteğe bağlı özellikler |
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
Çıkış <U> | Çıkış olarak () Tensörün sembolik tutamacını döndürür. |
static <U TNumber'ı genişletir, T TNumber'ı genişletir > CudnnRnnParamsSize <U> | |
statik CudnnRnnParamsSize.Options | yön (Dize yönü) |
statik CudnnRnnParamsSize.Options | bırakma (Kayanma bırakma) |
statik CudnnRnnParamsSize.Options | inputMode (Dize girişModu) |
statik CudnnRnnParamsSize.Options | numProj (Uzun numProj) |
Çıkış <U> | |
statik CudnnRnnParamsSize.Options | rnnMode (Dize rnnMode) |
statik CudnnRnnParamsSize.Options | tohum (Uzun tohum) |
statik CudnnRnnParamsSize.Options | tohum2 (Uzun tohum2) |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Genel Yöntemler
genel Çıkış <U> asOutput ()
Tensörün sembolik tutamacını döndürür.
TensorFlow işlemlerinin girdileri, başka bir TensorFlow işleminin çıktılarıdır. Bu yöntem, girişin hesaplanmasını temsil eden sembolik bir tanıtıcı elde etmek için kullanılır.
public static CudnnRnnParamsSize <U> create ( Kapsam kapsamı, İşlenen < TInt32 > numLayers, İşlenen < TInt32 > numUnits, İşlenen < TInt32 > inputSize, Class<T> T, Class<U> S, Seçenekler... seçenekler)
Yeni bir CudnnRnnParamsSize işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- CudnnRnnParamsSize'ın yeni bir örneği