I\'m trying to build a model with a custom loss. The loss needs to get additional the input for the model so my setup looks (a little bit unconventional) like this: