How To Create This Custom Loss Function In Keras And Ensure It Is Differentiable?

I am using the functional Model API and I would like to create the following custom loss function in Keras: LOSS EQUATION IMAGE I am implementing an unsupervised CNN and the value

Solution 1:

Solved.

Variables U and V should be created as follows:

U = model.output[0, :, :, 0]
V = model.output[0, :, :, 1]