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

LmyQAQ

于 2022-10-03 20:13:09 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

文章标签： python 深度学习人工智能

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.youkuaiyun.com/ws_syydr/article/details/127155706

本文详细解析了PyTorch中线性层nn.Linear和卷积层nn.Conv2d的输入张量维度要求。对于nn.Linear层，输入张量的最后一维必须等于该层的输入特征数in_features；对于nn.Conv2d层，输入张量的第二维必须等于该层的输入通道数in_channels。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

对于某需要输入到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

评论 2

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。