Custom loss function - Invalid argument: Expected begin and size arguments to be 1-D tensors of size 0, but got shapes [2] and [2] instead

Question

I am trying to apply a custom loss function to my CNN. This loss function should take in the prediction(event) and some "input variables" from that event. The CNN outp