ws_syydr的博客

原创 nn.Linear()、nn.Conv2d()的输入维度与输出维度的问题

对于某需要输入到nn.Linear(in_features, out_features)中的tensor[a, b, c, d, e]需要满足：e == in_features。对于某需要输入到nn.Conv2d(in_channels, out_channels)的中的tensor[a, b, c, d]需要满足：b == in_channels。