CudnnRNNParamsToCanonicalV2

パブリック最終クラスCudnnRNNParamsToCanonicalV2

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>
create (スコープscope、オペランド<Integer> numLayers、オペランド<Integer> numUnits、オペランド<Integer> inputSize、オペランド<T> params、Long numParamsWeights、Long numParamsBiases、 Options... options)
新しい CudnnRNNParamsToCanonicalV2 オペレーションをラップするクラスを作成するファクトリ メソッド。
静的CudnnRNNParamsToCanonicalV2.Options
方向(弦方向)
静的CudnnRNNParamsToCanonicalV2.Options
ドロップアウト(フロートドロップアウト)
静的CudnnRNNParamsToCanonicalV2.Options
inputMode (文字列入力モード)
静的CudnnRNNParamsToCanonicalV2.Options
numProj (長い numProj)
静的CudnnRNNParamsToCanonicalV2.Options
rnnMode (文字列 rnnMode)
静的CudnnRNNParamsToCanonicalV2.Options
種子(長い種子)
静的CudnnRNNParamsToCanonicalV2.Options
シード2 (ロングシード2)
リスト<出力<T>>

継承されたメソッド

パブリックメソッド

public List<出力<T>>バイアス()

public static CudnnRNNParamsToCanonicalV2 <T> create (スコープscope、オペランド<Integer> numLayers、オペランド<Integer> numUnits、オペランド<Integer> inputSize、オペランド<T> params、Long numParamsWeights、Long numParamsBiases、 Options...オプション)

新しい CudnnRNNParamsToCanonicalV2 オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
オプションオプションの属性値を持ちます
戻り値
  • CudnnRNNParamsToCanonicalV2 の新しいインスタンス

public static CudnnRNNParamsToCanonicalV2.Options方向(文字列方向)

public static CudnnRNNParamsToCanonicalV2.Optionsドロップアウト(Float ドロップアウト)

public static CudnnRNNParamsToCanonicalV2.Options inputMode (String inputMode)

public static CudnnRNNParamsToCanonicalV2.Options numProj (Long numProj)

public static CudnnRNNParamsToCanonicalV2.Options rnnMode (String rnnMode)

public static CudnnRNNParamsToCanonicalV2.Optionsシード(ロングシード)

public static CudnnRNNParamsToCanonicalV2.Optionsシード 2 (ロング シード 2)

public List<出力<T>>重み()