本文详细解析了在PyTorch中遇到的RuntimeError：两个矩阵维度不匹配的问题。通过实例展示了如何将0D和1D张量转换为适合矩阵乘法的形状，使用unsqueeze和permute方法确保维度正确匹配。了解这一技巧对于进行深度学习模型的搭建和训练至关重要。
out = torch.matmul(alpha,V)
报错：RuntimeError: both arguments to matmul need to be at least 1D, but they are 0D and 1D
原因：print alpha.shape, 结果为 torch.Size([]), 是一个0d tensor。Matrix multiplication 需要 1d tensor。
解决： unsqueeze输入使其维度匹配：out =torch.matmul(alpha.unsqueeze(dim=0).permute(1,0), V.unsqueeze(dim=0))
362
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
