2021-02-01

import torch
X = torch.tensor([[1.,0,0],[1,1,0],[1,0,1],[1,1,1]],dtype = torch.float32)
w = torch.tensor([-0.2,0.15,0.15] ,dtype = torch.float32) # b,w1,w2
z = torch.tensor([-0.2, -0.05, -0.05,  0.1] ,dtype = torch.float32)
def LineaR(X,w):
    zhat = torch.mv(X,w)
    return zhat
zhat = LineaR(X,w)
torch.allclose(zhat,z)

注意不要踩坑哦！！！
精度的问题，提高精度，可使用dtype = torch.float64