Pytorch Cross Product Definition