in utils.py [0:0]
def tf_print(t, name, summarize=10, first_n=None, mv=False, maxmin=False):
# Useful for debugging!
axes = [i for i in range(len(t.shape))]
if mv:
m, v = tf.nn.moments(t, axes=axes)
if maxmin:
maxi = tf.reduce_max(t)
mini = tf.reduce_min(t)
prefix = f'{tf.get_variable_scope().name}-{name}'
with tf.device('/cpu:0'):
if mv:
t = tf.Print(t, [tf.shape(t), m, v], prefix,
summarize=summarize, first_n=first_n)
elif maxmin:
t = tf.Print(t, [tf.shape(t), mini, maxi, t], prefix,
summarize=summarize, first_n=first_n)
else:
t = tf.Print(t, [tf.shape(t), t], prefix,
summarize=summarize, first_n=first_n)
return t