rnn = nn.LSTMCell(10, 20)
input = Variable(torch.randn(6, 3, 10))
hx = Variable(torch.randn(3, 20))
cx = Variable(torch.randn(3, 20))
output = []
for i in range(6):
…     hx, cx = rnn(input, (hx, cx))
…     output.append(hx)



In the above code, in for loop, i think input should be changed to input[i].
If i’m right, example code for  torch.nn.LSTMCell and torch.nn.GRUCell should be changed.
The example codes are here. http://pytorch.org/docs/nn.html#recurrent-layers 9