https://lacucinadiadine.blogspot.com/2024/03/writing-custom-loss-function-element-by.html