Printing tensor values with tf.Print()