目录 1. nn.LSTM2. nn.LSTMCell 1. nn.LSTM 初始化函数输入参数与RNN相同,分别是input_size,hidden_size和num_layer foward函数也与RNN类似,只不过返回值除了out外,ht变为(ht,ct) 代码见下图: 2. nn.LSTMCell 初始化函数输入参数与nn.LSTM类似 forward函数与nn.LSTM的区别是没有out 单层代码如下图: 和RNN的Cell类似,Cell单元循环更新h,c 2层无非就是多创建了一个cell2、h2、c2,并同时循环更新h1,c1,h2,c2,多层以此类推。