https://catalogcasiokeyboardlk.blogspot.com/2024/01/pytorch-custom-loss-function.html