Implement axis version of numpy.roll