LSTMBlockCell

パブリック最終クラスLSTMBlockCell

1 タイム ステップの LSTM セル順方向伝播を計算します。

この実装では 1 つの重み行列と 1 つのバイアス ベクトルを使用し、オプションののぞき穴接続があります。

このカーネル演算は、次の数式を実装します:

xh = [x, h_prev]
 [i, f, ci, o] = xh * w + b
 f = f + forget_bias
 
 if not use_peephole:
   wci = wcf = wco = 0
 
 i = sigmoid(cs_prev * wci + i)
 f = sigmoid(cs_prev * wcf + f)
 ci = tanh(ci)
 
 cs = ci .* i + cs_prev .* f
 cs = clip(cs, cell_clip)
 
 o = sigmoid(cs * wco + o)
 co = tanh(cs)
 h = co .* o
 

ネストされたクラス

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

パブリックメソッド

静的LSTMBlockCell.Options
cellClip (Float cellClip)
出力<T>
()
セル入力。
出力<T>
()
タン後のセル。
static <T extends Number> LSTMBlockCell <T>
create (スコープscope,オペランド<T> x,オペランド<T> csPrev,オペランド<T> hPrev,オペランド<T> w,オペランド<T> wci,オペランド<T> wcf,オペランド<T> wco,オペランド<T > b、オプション...オプション)
新しい LSTMBlockCell オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
cs ()
タンニング前のセルの状態。
出力<T>
f ()
忘れの門。
静的LSTMBlockCell.Options
rememberBias (浮動小数点忘れバイアス)
出力<T>
()
出力 h ベクトル。
出力<T>
()
入力ゲート。
出力<T>
ああ()
出力ゲート。
静的LSTMBlockCell.Options
usePeephole (ブール型 usePeephole)

継承されたメソッド

パブリックメソッド

public static LSTMBlockCell.Options cellClip (Float cellClip)

パラメーター
セルクリップ「cs」値をクリップする値。

public出力<T> ci ()

セル入力。

public出力<T> co ()

タン後のセル。

public static LSTMBlockCell <T> create (スコープscope,オペランド<T> x,オペランド<T> csPrev,オペランド<T> hPrev,オペランド<T> w,オペランド<T> wci,オペランド<T> wcf,オペランド<T > wco、オペランド<T> b、オプション...オプション)

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

パラメーター
範囲現在のスコープ
バツLSTM セルへの入力、形状 (batch_size、num_inputs)。
cs前前のタイムステップでのセル状態の値。
h前へ前のタイムステップでの前のセルの出力。
w重み行列。
WCI入力ゲートののぞき穴接続の重み行列。
wcf Forget Gate Peephole 接続の重み行列。
ウィコ出力ゲートのピープホール接続の重み行列。
bバイアスベクトル。
オプションオプションの属性値を持ちます
戻り値
  • LSTMBlockCell の新しいインスタンス

public出力<T> cs ()

タンニング前のセルの状態。

public出力<T> f ()

忘れの門。

public static LSTMBlockCell.Options rememberBias (浮動小数点忘れバイアス)

パラメーター
バイアスを忘れる忘却ゲートバイアス。

public出力<T> h ()

出力 h ベクトル。

public出力<T> i ()

入力ゲート。

public出力<T> o ()

出力ゲート。

public static LSTMBlockCell.Options usePeephole (ブール型 usePeephole)

パラメーター
のぞき穴を使用するのぞき穴ウェイトを使用するかどうか。