https://discountsnowblowersthisinstant.blogspot.com/2022/10/pytorch-custom-loss-function.html