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 op 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 | Los atributos opcionales para LSTMBlockCell |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estáticas 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 se extiende TNumber > LSTMBlockCell <T> | |
Salida <T> | cs () El estado celular antes del tanh. |
Salida <T> | f () La puerta del olvido. |
estáticas LSTMBlockCell.Options | forgetBias (forgetBias float) |
Salida <T> | h () El vector de salida h. |
Salida <T> | i () La puerta de entrada. |
Salida <T> | o () La puerta de salida. |
estáticas LSTMBlockCell.Options | usePeephole (Boolean usePeephole) |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
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> crear ( Alcance alcance, operando <T> x, operando <T> csPrev, operando <T> hPrev, operando <T> w, operando <T> wci, operando <T> WCF, operando <T > OMA, 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
estáticas pública LSTMBlockCell.Options forgetBias (forgetBias Float)
Parámetros
olvidarBias | El sesgo de la puerta del olvido. |
---|
public static LSTMBlockCell.Options usePeephole (booleano usePeephole)
Parámetros
utilizar mirilla | Ya sea para usar pesas de mirilla. |
---|