Thank you, albanD.
Actually, I use this kind of tensor (sparsetensor.to_dense().sum()) with torch.nn.Linear.
I’m not sure whether torch.nn.Linear use einsum but error with torch.nn.Linear might be the same problem (lack of support for sparse tensor).
I’ll open a feature request soon.