CudnnRNN パラメータを正規形式で取得します。 LSTM での投影をサポートします。
不透明な params バッファから重みのセットを取得します。これは、将来の実行と互換性のある方法で保存および復元できます。
params バッファーは、異なる GPU 間で互換性がない可能性があることに注意してください。したがって、保存と復元は正規の重みとバイアスとの間で変換する必要があります。
num_layers: RNN モデルの層の数を指定します。 num_units: 非表示状態のサイズを指定します。 input_size: 入力ステートのサイズを指定します。 num_params_weights: すべてのレイヤーの重みパラメータ行列の数。 num_params_biases: すべてのレイヤーのバイアス パラメーター ベクトルの数。重み: 保存と復元に使用できる重みの標準形式。異なる世代間でも互換性がある可能性が高くなります。バイアス: 保存と復元に使用できるバイアスの標準形式。異なる世代間でも互換性がある可能性が高くなります。 rnn_mode: RNN モデルのタイプを示します。 input_mode: 入力と最初の層の前の実際の計算の間に線形投影があるかどうかを示します。 「skip_input」は、input_size == num_units の場合にのみ許可されます。 input_size == num_units の場合、「auto_select」は「skip_input」を意味します。それ以外の場合は、「linear_input」を意味します。方向: 双方向モデルが使用されるかどうかを示します。 dir = (方向 == 双方向) ? 2 : 1 ドロップアウト: ドロップアウトの確率。 0.に設定すると、ドロップアウトは無効になります。シード: ドロップアウトを初期化するシードの最初の部分。シード2: ドロップアウトを初期化するためのシードの 2 番目の部分。 num_proj: 射影行列の出力次元。 None または 0 の場合、投影は実行されません。
ネストされたクラス
クラス | CudnnRNNParamsToCanonicalV2.Options | CudnnRNNParamsToCanonicalV2 のオプションの属性 |
パブリックメソッド
リスト<出力<T>> | 偏見() |
static <T extends Number> CudnnRNNParamsToCanonicalV2 <T> | |
静的CudnnRNNParamsToCanonicalV2.Options | 方向(弦方向) |
静的CudnnRNNParamsToCanonicalV2.Options | ドロップアウト(フロートドロップアウト) |
静的CudnnRNNParamsToCanonicalV2.Options | inputMode (文字列入力モード) |
静的CudnnRNNParamsToCanonicalV2.Options | numProj (長い numProj) |
静的CudnnRNNParamsToCanonicalV2.Options | rnnMode (文字列 rnnMode) |
静的CudnnRNNParamsToCanonicalV2.Options | 種子(長い種子) |
静的CudnnRNNParamsToCanonicalV2.Options | シード2 (ロングシード2) |
リスト<出力<T>> | 重み() |
継承されたメソッド
パブリックメソッド
public static CudnnRNNParamsToCanonicalV2 <T> create (スコープscope、オペランド<Integer> numLayers、オペランド<Integer> numUnits、オペランド<Integer> inputSize、オペランド<T> params、Long numParamsWeights、Long numParamsBiases、 Options...オプション)
新しい CudnnRNNParamsToCanonicalV2 オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を持ちます |
戻り値
- CudnnRNNParamsToCanonicalV2 の新しいインスタンス