1、创建一个全连接层
model = torch.nn.Linear(1, 1)
2、传入数据,model作为可调用对象被当做函数调用,并输出返回值y_predict
y_predict = model(x_data)
3、源码
class Module(object):
# 当子类被当做可调用对象调用时,会调用其重写的forward方法
def __call__(self, *input, **kwargs):
result = self.forward(*input, **kwargs)
class Linear(Module):
# 构造函数时,初始化参数权重矩阵w和偏置b
def __init__(self, in_features, out_features, bias