Are you perhaps thinking of torch.matmul? torch.mul (and torch.multiply, and the * operator) perform element-wise multiplication.

torch.matmul was one of the first functions I implemented on WhyTorch and it uses and highlights rows and columns as you would expect.

I’d love to hear any feedback or outcomes from your training session, please feel free to reach out - email in profile.