Torch linalg pinv