Cudnn RNN モデルで使用できる重みのサイズを計算します。
Cudnn RNN モデルで使用できる params サイズを返します。後続の重みの割り当てと初期化では、このサイズを使用する必要があります。
num_layers: RNN モデルの層の数を指定します。 num_units: 非表示状態のサイズを指定します。 input_size: 入力ステートのサイズを指定します。 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 番目の部分。 params_size: この RNN モデルに割り当てて初期化する必要がある params バッファーのサイズ。この params バッファーは GPU 間で互換性がない可能性があることに注意してください。 CudnnRNNParamsWeights および CudnnRNNParamsBiases を使用して、異なる実行間で互換性のある方法で保存および復元してください。
ネストされたクラス
クラス | CudnnRnnParamsSize.Options | CudnnRnnParamsSize のオプションの属性 |
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<U> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <U extends TNumber , T extends TNumber > CudnnRnnParamsSize <U> | |
静的CudnnRnnParamsSize.Options | 方向(弦方向) |
静的CudnnRnnParamsSize.Options | ドロップアウト(フロートドロップアウト) |
静的CudnnRnnParamsSize.Options | inputMode (文字列入力モード) |
静的CudnnRnnParamsSize.Options | numProj (長い numProj) |
出力<U> | paramsSize () |
静的CudnnRnnParamsSize.Options | rnnMode (文字列 rnnMode) |
静的CudnnRnnParamsSize.Options | 種子(長い種子) |
静的CudnnRnnParamsSize.Options | シード2 (ロングシード2) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
パブリックメソッド
public Output <U> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static CudnnRnnParamsSize <U> create (スコープスコープ、オペランド< TInt32 > numLayers、オペランド< TInt32 > numUnits、オペランド< TInt32 > inputSize、Class<T> T、Class<U> S、オプション...オプション)
新しい CudnnRnnParamsSize オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を持ちます |
戻り値
- CudnnRnnParamsSize の新しいインスタンス