torch.einsum doesn't accept float tensors?
I get the following error:

RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #3 'mat2' in call to _th_addmm_out

I use torch.einsum as follows:
mu = torch.einsum('ijl, akij -&gt; akl', idxs, activation_map)
I don't understand this, as in the documentation they are using float tensors too (https://pytorch.org/docs/stable/generated/torch.einsum.html). Also choosing a long tensor is no option, as all values in activation_map are between 0 and 1.

It seems like your first argument, idxs is of type Long.
All input tensors to torch.einsum should be Float.
