Implement axis version of ``numpy.roll''