https://www.wwwinsights.com/ai/pytorch-own-tensor-not-numpy/