目录
- [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,多层以此类推。