TypeError: einsum() takes 2 positional arguments but 3 were given

torch版本问题,torch>1.0可以写为torch.einsum(‘ik,jk->ij’,a,b).
torch<1.0需要写为torch.einsum(‘ik,jk->ij’,[a,b]).

你可能感兴趣的:(pyrhon)