Custom Loss Function Tensorflow 2