torch版本问题,torch>1.0可以写为torch.einsum(‘ik,jk->ij’,a,b).
torch<1.0需要写为torch.einsum(‘ik,jk->ij’,[a,b]).
TypeError: einsum() takes 2 positional arguments but 3 were given
最新推荐文章于 2024-09-04 08:22:11 发布
torch版本问题,torch>1.0可以写为torch.einsum(‘ik,jk->ij’,a,b).
torch<1.0需要写为torch.einsum(‘ik,jk->ij’,[a,b]).