https://videosashka.com/watch/SvgMYjozLE8/convert-pytorch-tensor-to-numpy