Pytorch matrix multiplication with 2nd matrix repeated I’m trying to do matrix multiplication in torch. It goes like this