TensorFlow实现了自己的log输出模块:tf.logging,logging一共有五个日志级别,从低到高分别为:
tf.logging.DEBUG
tf.logging.INFO
tf.logging.WARN
tf.logging.ERROR
tf.logging.FATAL
当使用tf.logging.set_verbosity(tf.logging.DEBUG)设定日志级别为DEBUG级别时,所有的logging输出都会被打印到屏幕上,
当使用tf.logging.set_verbosity(tf.logging.INFO)设定日志级别为INFO级别时,只有INFO级别及以上的logging会被打印到屏幕上,
当使用tf.logging.set_verbosity(tf.logging.WARN)设定日志级别为WARN级别时,只有WARN级别及以上的logging会被打印到屏幕上,
当使用tf.logging.set_verbosity(tf.logging.ERROR)设定日志级别为ERROR级别时,只有ERROR级别及以上的logging会被打印到屏幕上,
当使用tf.logging.set_verbosity(tf.logging.FATAL)设定日志级别为FATAL级别时,只有FATAL级别及以上的logging会被打印到屏幕上。
具体实现可以参考下面的代码:
import tensorflow as tf
import numpy as np
tf.logging.set_verbosity(tf.logging.DEBUG)
#tf.logging.set_verbosity(tf.logging.INFO)
#tf.logging.set_verbosity(tf.logging.WARN)
#tf.logging.set_verbosity(tf.logging.ERROR)
#tf.logging.set_verbosity(tf.logging.FATAL)
tf.logging.debug('Test tf logging output image size: %dx%d' % (100, 100))
tf.logging.info('Test tf logging output image size: %dx%d' % (100, 100))
tf.logging.warn('Test tf logging output image size: %dx%d' % (100, 100))
tf.logging.error('Test tf logging output image size: %dx%d' % (100, 100))
tf.logging.fatal('Test tf logging output image size: %dx%d' % (100, 100))
a=np.array([[1,0,0],[0,1,1]])
a1=np.array([[3,2,3],[4,5,6]])
equal_one = tf.equal(a,1)
equal_one_index = tf.where(equal_one)
new_al = tf.where(tf.equal(a,1),a1,1-a1)
with tf.Session() as sess:
print('equal_one --------------------------')
print(equal_one.eval())
print('equal_one_index --------------------------')
print(equal_one_index.eval())
print('new_al --------------------------')
print(new_al.eval())