对到来的数据使用线性变换:
参数:
形状:
其中∗*∗ 的含义为附加维的任何数值,并且
, 除了最后一个维度外,其他所有维度都与输入相同,并且
.
变量:
得到的 初始化的值,其中
,其中
例:
>>> m = nn.Linear(20, 30) >>> input = torch.randn(128, 20) >>> output = m(input) >>> print(output.size()) torch.Size([128, 30])