https://videosashka.com/watch/2yBEZzQu8dA/tensors-with-pytorch-deep-learning-with-pytorch-2