CNN with LSTM for ND with Pytorch