https://yuridekim.github.io//posts/Torch-adding-custom-loss-functions/