https://www.wwwinsights.com/ai/torch-tensor-to-represent-images/