首页 > 技术文章 > 1、第一个TensorFlow程序

pengzhonglian 2019-11-11 16:59 原文

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #去掉警告,将警告级别提升

a = tf.constant(2)   #定义一个常量
b = tf.constant(4)
sum = tf.add(a,b)    #加法操作

with tf.Session() as sess:  #上下文管理
    print(sess.run(sum))   #run运行加法op

输出:

6

 

推荐阅读