先贴官方文档
nn.Linear是完成从in_features到out_features的线性变换。实例化完成后input的大小可以有多维,但最后一维的大小必须和in_features一致。
>>> m = nn.Linear(20, 30)
>>> input = torch.randn(128, 20)
>>> output = m(input)
>>> print(output.size())
torch.Size([128, 30])
先贴官方文档
nn.Linear是完成从in_features到out_features的线性变换。实例化完成后input的大小可以有多维,但最后一维的大小必须和in_features一致。
>>> m = nn.Linear(20, 30)
>>> input = torch.randn(128, 20)
>>> output = m(input)
>>> print(output.size())
torch.Size([128, 30])