pytorch-pytorch之LSTM

目录

• [1. nn.LSTM](#1. nn.LSTM)
• [2. nn.LSTMCell](#2. nn.LSTMCell)

1. nn.LSTM

初始化函数输入参数与RNN相同，分别是input_size，hidden_size和num_layer

foward函数也与RNN类似，只不过返回值除了out外，h_t变为(h_t,c_t)

代码见下图：

2. nn.LSTMCell

初始化函数输入参数与nn.LSTM类似

forward函数与nn.LSTM的区别是没有out

单层代码如下图：

和RNN的Cell类似，Cell单元循环更新h,c

2层无非就是多创建了一个cell2、h2、c2，并同时循环更新h1,c1,h2,c2，多层以此类推。