Pytorch Cross Product