在tensorflow中绘制一条线

import tensorflow as tf
import matplotlib.pyplot as plt

# 创建输入数据
x = tf.linspace(-1.0, 1.0, 100)
# 定义线性函数 y = 2x + 1
y = 2 * x + 1

# 创建会话
with tf.Session() as sess:
    # 执行计算
    x_vals, y_vals = sess.run([x, y])

# 绘制线条
plt.plot(x_vals, y_vals, 'r-', label='y=2x+1')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Linear Function')
plt.legend(loc='best')

# 显示图像
plt.show()