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 の場合、投影は実行されません。
ネストされたクラス
クラス | CudnnRNNParamsToCanonical.Options | CudnnRNNParamsToCanonical のオプション属性 |
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
リスト<出力<T>> | 偏見() |
static <T extends TNumber > CudnnRNNParamsToCanonical <T> | |
静的CudnnRNNParamsToCanonical.Options | 方向(弦方向) |
静的CudnnRNNParamsToCanonical.Options | ドロップアウト(フロートドロップアウト) |
静的CudnnRNNParamsToCanonical.Options | inputMode (文字列入力モード) |
静的CudnnRNNParamsToCanonical.Options | numProj (長い numProj) |
静的CudnnRNNParamsToCanonical.Options | rnnMode (文字列 rnnMode) |
静的CudnnRNNParamsToCanonical.Options | 種子(長い種子) |
静的CudnnRNNParamsToCanonical.Options | シード2 (ロングシード2) |
リスト<出力<T>> | 重み() |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
パブリックメソッド
public static CudnnRNNParamsToCanonical <T> create (スコープスコープ、オペランド< TInt32 > numLayers、オペランド< TInt32 > numUnits、オペランド< TInt32 > inputSize、オペランド<T> params、Long numParamsWeights、Long numParamsBiases、 Options...オプション)
新しい CudnnRNNParamsToCanonical オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を持ちます |
戻り値
- CudnnRNNParamsToCanonical の新しいインスタンス