https://keroseneheaterreview.blogspot.com/2023/08/pytorch-equivalent-to.html