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の場合にのみ許可されます。 'auto_select'は、input_size == num_unitsの場合、 'skip_input'を意味します。それ以外の場合は、「linear_input」を意味します。方向:双方向モデルを使用するかどうかを示します。 dir =(方向==双方向)? 2:1ドロップアウト:ドロップアウト確率。 0に設定すると、ドロップアウトは無効になります。シード:ドロップアウトを初期化するシードの最初の部分。 seed2:ドロップアウトを初期化するシードの2番目の部分。 num_proj:射影行列の出力次元。 Noneまたは0の場合、投影は実行されません。
ネストされたクラス
クラス | CudnnRNNParamsToCanonicalV2.Options | CudnnRNNParamsToCanonicalV2 オプションの属性 |
パブリックメソッド
リスト<出力<T >> | バイアス() |
static <T extends Number> CudnnRNNParamsToCanonicalV2 <T> | |
静的CudnnRNNParamsToCanonicalV2.Options | 方向(文字列方向) |
静的CudnnRNNParamsToCanonicalV2.Options | ドロップアウト(フロートドロップアウト) |
静的CudnnRNNParamsToCanonicalV2.Options | inputMode (String inputMode) |
静的CudnnRNNParamsToCanonicalV2.Options | numProj (長いnumProj) |
静的CudnnRNNParamsToCanonicalV2.Options | rnnMode (文字列rnnMode) |
静的CudnnRNNParamsToCanonicalV2.Options | 種子(ロングシード) |
静的CudnnRNNParamsToCanonicalV2.Options | seed2 (ロングシード2) |
リスト<出力<T >> | 重み() |
継承されたメソッド
パブリックメソッド
public static CudnnRNNParamsToCanonicalV2 <T> create (スコープスコープ、オペランド<整数> numLayers、オペランド<整数> numUnits、オペランド<整数> inputSize、オペランド<T>パラメータ、ロングnumParamsWeights、ロングnumParamsBiases、オプション...オプション)
新しいCudnnRNNParamsToCanonicalV2操作をラップするクラスを作成するファクトリメソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を運ぶ |
戻り値
- CudnnRNNParamsToCanonicalV2の新しいインスタンス