I am trying to a custom loss function for tensorflow. Because I was trying to see what was going on in the loss function, I tried to print the input of the loss function as