torch.einsum 简单介绍计算流程
torch.einsum>>> a = torch.arange(60.).reshape(5,3,4)>>> b = torch.arange(24.).reshape(3,4,2)>>> o = torch.einsum('fnd,ndh->fh', a, b)>>> otensor([[1012., 1078.
