热门
最新
红包
立Flag
投票
同城
我的
发布
【报错】 s_local = torch.mm(s, A)
RuntimeError: self must be a matrix。在数学上矩阵是二维的,这里的torch.mm使用范围比较狭窄,必须是二维。而这里的s是[B,N,F]三维,A是[N,N]二维的。结果要变为[B,N,F],因此修改为torch.matmul(A,s);注意不是 torch.matmul(s,A).这里有顺序的区别
torch中的矩阵乘法,可以看连接:https://blog.csdn.net/mimiduck/article/details/119797012
CSDN App 扫码分享
评论
点赞
打赏
- 复制链接
- 举报