Pytorch Matrix Operations