Solved by KFrank in post #2 
              
              
                Hi Masilive! 

The short answer is that you have the dimensions of your input 
swapped.  You would want something like, for example, “m1: [10 x 5]”. 

Pytorch will interpret state as a batch of 5 samples, each of which 
is a vector of 10 values. 

But the first layer of your Sequential is a Linear t…