torch.nn.Linear 用法含义

input = torch.randn(128, 20) # 输入特征维度为（128，20）。
m = nn.Linear(20, 30) #构建pytorch中神经网络的（nn）的线性层，且输入为20个特征，输出为30个特征。
output = m(input) # output结果为将input进行线性处理。
print(output.size())

输出结果为：（128，30）

如上代码， 输入特征维度为（128，20）

构建pytorch中神经网络的（nn）的线性层，且输入为20个特征，输出为30个特征

output = 将input进行线性处理。

于是得到输出为（128，30）

注意： 代码中有两个20，这两个数字必须要相同。如果你改为一个10 ，一个20 ，则程序报错error。

具体可参考网址

https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-nn/