pytorch nn.Linear的用法和理解

先贴官方文档
nn.Linear是完成从in_features到out_features的线性变换。实例化完成后input的大小可以有多维,但最后一维的大小必须和in_features一致。
pytorch nn.Linear的用法和理解_第1张图片
pytorch nn.Linear的用法和理解_第2张图片

>>> m = nn.Linear(20, 30)
>>> input = torch.randn(128, 20)
>>> output = m(input)
>>> print(output.size())
torch.Size([128, 30])

你可能感兴趣的:(PyTorch,pytorch,深度学习,机器学习)