Convert a function to Tensorflow loss

Question

I created the following custom loss function, that I would like to convert to a Tensorflow-compatible one:

def custom_loss(y_train, f_prediction):
    
    ab