sorry, I think I have expressed myself poorly.
I do not mean that the calculation is slower. In fact einsum is faster (around 25% on average, I believe). My problem is that on average, the accuracy of my predictions (it’s a classification task), is slightly worse.