Pytorch Cross Product Example