pytorch api torch.nn.Linear

线性层实现了 y = x A T + b y=xA^T+b y=xAT+b

参数 描述
in_features 输入节点数
out_features 输出节点数
bias 如果为假,将不会有b。默认为真
对象 描述
self.in_features in_features
self.out_features out_features
self.weight Parameter(torch.Tensor(out_features, in_features))
self.bias Parameter(torch.Tensor(out_features))
forward(self, input) F.linear(input, self.weight, self.bias)

参考文献:
https://pytorch.org/docs/stable/nn.html#linear-layers

你可能感兴趣的:(人工神经网络)