pytroch学习记录三维绘图plot_surface

绘制三维图
流程：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
#必要的库
x_data = [1.0, 2.0, 3.0]
y_data = [2.0, 4.0, 6.0]
#这个是一个线性模型y=2x，后面计算损失函数

def forward(x):
    return x * w + b
#返回预测值

def loss(x, y):
    y_pred = forward(x)
    return (y_pred - y) ** 2
#损失函数

W = np.arange(0.0, 4.1, 0.1)
B = np.arange(-2.0, 2.1, 0.1)
[w, b] = np.meshgrid(W, B)
#meshgrid为网格剖分


l_sum = 0
for x_val, y_val in zip(x_data, y_data):
    y_pred_val = forward(x_val)
    loss_val = loss(x_val, y_val)
    l_sum = l_sum + loss_val
#注意这里的l_sum并不是一个数字，而是一个w*b的二维数组，
# 因为在这个for循环里，w,b本身就是数组，所以得到的l_sum就是一个二位数组
#注意了这种参数w，b是数组那么得到的结果也是数组


fig = plt.figure()
ax = Axes3D(fig)
#固定格式？？？
ax.plot_surface(w,b, l_sum/3)
#注意w,b,是前面的[w, b] = np.meshgrid(W, B)
#l_sum/3是一个二维数组，注意了
plt.show()