https://keroseneheaterreview.blogspot.com/2024/02/how-do-i-modify-this-pytorch.html