パブリック最終クラス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> | |
出力<T> | cs () タンニング前のセルの状態。 |
出力<T> | f () 忘れの門。 |
静的LSTMBlockCell.Options | rememberBias (浮動小数点忘れバイアス) |
出力<T> | ひ() 出力 h ベクトル。 |
出力<T> | 私() 入力ゲート。 |
出力<T> | ああ() 出力ゲート。 |
静的LSTMBlockCell.Options | usePeephole (ブール型 usePeephole) |
継承されたメソッド
パブリックメソッド
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 | 重み行列。 |
ウィスコンシン州 | 入力ゲートののぞき穴接続の重み行列。 |
wcf | Forget Gate Peephole 接続の重み行列。 |
ウィコ | 出力ゲートのピープホール接続の重み行列。 |
b | バイアスベクトル。 |
オプション | オプションの属性値を持ちます |
返品
- LSTMBlockCell の新しいインスタンス