美文网首页
RuntimeError: Given groups=1, we

RuntimeError: Given groups=1, we

作者: pure_joy | 来源:发表于2023-10-27 15:26 被阅读0次
遇到问题:缝合两个网络模型时,输入维度不一致导致模型,训练报错。
解决思路:先用troch.flatten()函数把tensor张量展开成一维张量,在使用[:xx]截取需要维度元素总数,最后使用view()函数重塑成需要的张量维度。

相关文章

网友评论

      本文标题:RuntimeError: Given groups=1, we

      本文链接:https://www.haomeiwen.com/subject/cltpidtx.html