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/