I should be careful and not mix in any numpy functions in the pytorch functions, right? I bet pytorch has all I need for fast matrix multiplication/manipulations?
Anyways, you’ve given me enough information and motivation to keep on trying and learning Pytorch! Thank you very much