https://capressocoffemakerthecheapest.blogspot.com/2024/02/pytorch-custom-loss-function.html