微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

tensorflow源码分析八- 打印log

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '0'
os.environ['TF_CPP_MIN_VLOG_LEVEL'] = '3'
from tensorflow import logging
logging.set_verbosity(logging.INFO)

with tf.device("/fpga:0"):
  result = tf.user_ops.vect_add(4, 0)
#  result = tf.user_ops.vect_add([4, 3], [2, 1])

#a = tf.constant([1.0, 2.0], name='a')
#b = tf.constant([2.0, 3.0], name='b')
#result = tf.add(a, b)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(result))

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐