神经网络的线性函数 wx+b=y import torch.nn as nn model = nn.Linear(2,1) input = torch.Tensor(1,2) output = model(input) <