Solved by KFrank in post #6 
              
              
                Hi Josep! 

I’m not sure what you are trying to do here. 
First, you need to package your input data as a pytorch tensor. 
(That’s what pytorch works with.) 
You need a batch size.  It can be 1, if you want to work with a single 
sample, but let’s say you use a batch size of 7.  Then a single batch 
…