3 ms·Looks like the two asterisks cancelled each other out in torch.sum((x @ y)**2)by mkl 6y agoLooks like the two asterisks cancelled each other out in torch.sum((x @ y)**2)