警告:このAPIは非推奨であり、置き換えが安定した後 TensorFlowの将来のバージョンで削除される予定です。

このページは Cloud Translation API によって翻訳されました。
Switch to English

CudnnRNNV3

パブリックファイナルクラスCudnnRNNV3

cuDNNに裏打ちされたRNN。

paramsバッファーに関して、入力状態と初期状態からRNNを計算します。 CudnnRNNよりも1つの追加入力「sequence_lengths」を受け入れます。

rnn_mode:RNNモデルのタイプを示します。 input_mode:入力と最初のレイヤーの前の実際の計算の間に線形射影があるかどうかを示します。 'skip_input'は、input_size == num_unitsの場合にのみ許可されます。 'auto_select'は、input_size == num_unitsの場合、 'skip_input'を意味します。それ以外の場合は、「linear_input」を意味します。方向:双方向モデルを使用するかどうかを示します。 「単方向」または「双方向」である必要があります。ドロップアウト:ドロップアウトの確率。 0に設定すると、ドロップアウトは無効になります。シード:ドロップアウトを初期化するシードの最初の部分。 seed2:ドロップアウトを初期化するシードの2番目の部分。 input:time_majorがtrueの場合、これは[seq_length、batch_size、input_size]の形状の3Dテンソルです。 time_majorがfalseの場合、形状は[batch_size、seq_length、input_size]です。 input_h:time_majorがtrueの場合、これは[num_layer * dir、batch_size、num_units]の形状の3Dテンソルです。 time_majorがfalseの場合、形状は[batch_size、num_layer * dir、num_units]です。 input_c:LSTMの場合、[num_layer * dir、batch、num_units]の形状の3Dテンソル。他のモデルの場合、無視されます。 params:不透明なレイアウトの重みとバイアスを含む1次元テンソル。サイズはCudnnRNNParamsSizeを使用して作成し、個別に初期化する必要があります。異なる世代間で互換性がない可能性があることに注意してください。したがって、各入力シーケンスの長さのベクトルであるsequence_lengthsを保存および復元することをお勧めします。出力:time_majorがtrueの場合、これは[seq_length、batch_size、dir * num_units]の形状の3Dテンソルです。 time_majorがfalseの場合、形状は[batch_size、seq_length、dir * num_units]です。 output_h:同じ形状にinput_hがあります。 output_c:LSTMのinput_cと同じ形状。他のモデルの空のテンソル。 is_training:この操作が推論またはトレーニングのどちらに使用されるかを示します。 time_major:入出力フォーマットがタイムメジャーかバッチメジャーかを示します。 reserved_space:バックプロパゲーション計算で使用できる不透明なテンソル。 is_trainingがtrueの場合にのみ生成されます。

ネストされたクラス

クラスCudnnRNNV3.Options CudnnRNNV3オプションの属性

パブリックメソッド

static <T extends Number> CudnnRNNV3 <T>
createスコープスコープ、オペランド<T>入力、オペランド<T> inputH、オペランド<T> inputC、オペランド<T>パラメータ、オペランド<整数> sequenceLengths、オプション...オプション)
新しいCudnnRNNV3操作をラップするクラスを作成するファクトリメソッド。
静的CudnnRNNV3.Options
方向(文字列方向)
静的CudnnRNNV3.Options
ドロップアウト(フロートドロップアウト)
出力<?>
hostReserved ()
静的CudnnRNNV3.Options
inputMode (String inputMode)
静的CudnnRNNV3.Options
isTraining (ブールisTraining)
静的CudnnRNNV3.Options
numProj (長いnumProj)
出力<T>
出力()
出力<T>
outputC ()
出力<T>
outputH ()
出力<T>
静的CudnnRNNV3.Options
rnnMode (文字列rnnMode)
静的CudnnRNNV3.Options
種子(ロングシード)
静的CudnnRNNV3.Options
seed2 (ロングシード2)
静的CudnnRNNV3.Options
timeMajor (ブールtimeMajor)

継承されたメソッド

パブリックメソッド

public static CudnnRNNV3 <T> create スコープスコープ、オペランド<T>入力、オペランド<T> inputH、オペランド<T> inputC、オペランド<T>パラメータ、オペランド<整数> sequenceLengths、オプション...オプション)

新しいCudnnRNNV3操作をラップするクラスを作成するファクトリメソッド。

パラメーター
範囲現在のスコープ
オプションオプションの属性値を運ぶ
戻り値
  • CudnnRNNV3の新しいインスタンス

public static CudnnRNNV3.Options direction (文字列方向)

public static CudnnRNNV3.Optionsドロップアウト(フロートドロップアウト)

public Output <?> hostReserved ()

public static CudnnRNNV3.Options inputMode (String inputMode)

public static CudnnRNNV3.Options isTraining (ブールisTraining)

public static CudnnRNNV3.Options numProj (Long numProj)

public Output <T> output ()

public Output <T> outputC ()

public Output <T> outputH ()

public Output <T> reservedSpace ()

public static CudnnRNNV3.Options rnnMode (String rnnMode)

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

public static CudnnRNNV3.Options seed2 (Long seed2)

public static CudnnRNNV3.Options timeMajor (ブールtimeMajor)