Keras custom loss function using outputs of each neuron

Question

I am trying to custom a loss function using the outputs of each neuron of the last layer. And the function may not be linear. Here is what I am working on:

##