¿Cómo puedo indexar el último eje de una matriz Numpy si no sé su rango de antemano?

Esto es lo que quiero hacer: Sea a una matriz Numpy de rango desconocido. Quiero la porción de los últimos k elementos del último eje.

Si a es 1D, quiero

 b = a[-k:]

Si a es 2D, quiero

 b = a[:, -k:]

Si a es 3D, quiero

 b = a[:, :, -k:]

y así.

Quiero que esto funcione independientemente del rango de a (siempre que el rango sea al menos 1).

El hecho de que quiera los últimos k elementos en el ejemplo es irrelevante, por supuesto, el punto es que quiero especificar índices para cualquiera que sea el último eje cuando no sé el rango de una matriz de antemano.