Computing jacobian determinant of a vector field

1

I have a vector field of shape [128, 128, 2], how do I compute the jacobian determinant of this vector field using torch.gradient?