clase final pública LSTMBlockCell
Calcula la propagación hacia adelante de la celda LSTM para 1 paso de tiempo.
Esta implementación usa 1 matriz de peso y 1 vector de sesgo, y hay una conexión de mirilla opcional.
Esta operación del núcleo implementa las siguientes ecuaciones matemáticas:
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
Clases anidadas
clase | LSTMBlockCell.Options | Atributos opcionales para LSTMBlockCell |
Métodos públicos
estático LSTMBlockCell.Options | cellClip (Float cellClip) |
Salida <T> | ci () La entrada de la celda. |
Salida <T> | co () La celda después del tanh. |
estática <T extiende Número> LSTMBlockCell <T> | |
Salida <T> | cs () El estado celular antes del tanh. |
Salida <T> | f () La puerta del olvido. |
estático LSTMBlockCell.Options | ForgetBias (Float ForgetBias) |
Salida <T> | h () El vector h de salida. |
Salida <T> | yo () La puerta de entrada. |
Salida <T> | o () La puerta de salida. |
estático LSTMBlockCell.Options | usePeephole (booleano usePeephole) |
Métodos heredados
Métodos públicos
public static LSTMBlockCell.Options cellClip (Float cellClip)
Parámetros
cellClip | Valor al que recortar el valor 'cs'. |
---|
public static LSTMBlockCell <T> create ( alcance de alcance, operando <T> x, operando <T> csPrev, operando <T> hPrev, operando <T> w, operando <T> wci, operando <T> wcf, operando <T > wco, operando <T> b, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LSTMBlockCell.
Parámetros
alcance | alcance actual |
---|---|
X | La entrada a la celda LSTM, forma (tamaño_de_lote, num_inputs). |
csPrev | Valor del estado de la celda en el paso de tiempo anterior. |
hPrev | Salida de la celda anterior en el paso de tiempo anterior. |
w | La matriz de peso. |
wci | La matriz de peso para la conexión de mirilla de la puerta de entrada. |
WC | La matriz de peso para la conexión de mirilla de puerta olvidada. |
wco | La matriz de peso para la conexión de mirilla de la puerta de salida. |
B | El vector de sesgo. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de LSTMBlockCell
LSTMBlockCell estática pública.Opciones olvidoBias (Flotante olvido)
Parámetros
olvidarBias | El sesgo de la puerta del olvido. |
---|
Public static LSTMBlockCell.Options usePeephole (Boolean usePeephole)
Parámetros
utilizar mirilla | Ya sea para usar pesas de mirilla. |
---|