I have a custom loss function in Keras that is returning nans and in order to debug it I want to print the values of my tensors. I have tried tf.print and