torch.einsum不接受浮动张量？

问题描述

我收到以下错误：

RuntimeError：标量类型为Long的预期对象，但调用_th_addmm_out的参数＃3'mat2'的标量类型为Float

我按如下方式使用torch.einsum：

mu = torch.einsum('ijl,akij -> akl',idxs,activation_map)

我不明白这一点，因为在文档中它们也使用浮动张量（https://pytorch.org/docs/stable/generated/torch.einsum.html）。另外，选择长张量也是没有选择的，因为activation_map中的所有值都在0到1之间。

解决方法

似乎您的第一个参数idxs的类型为Long。
torch.einsum的所有输入张量应为Float。