彻底解决Torch[mat1 dim 1 must match mat2 dim 0]
如果是卷积之后的层一定要先化为一个维度exp:x = x.view(x.size(),-1)ps:view函数的解释把原先tensor中的数据按照行优先的顺序排成一个一维的数据(这里应该是因为要求地址是连续存储的),然后按照参数组合成其他维度的tensor。比如说是不管你原先的数据是[[[1,2,3],[4,5,6]]]还是[1,2,3,4,5,6],因为它们排成一维向量都是6个元素,所以只要view后面的参数一致,得到的结果都是一样的。比如a=torch.Tensor([[[
原创
2022-02-21 16:55:54 ·
4555 阅读 ·
0 评论