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.
why is there a 'pass' here?