Add sort.grad support for vector and matrix when axis is None.

I'll close the other PR about this can't can't be easily rebased.