https://machinejuli.blogspot.com/2023/08/pytorch-basics-for-machine-learning.html