code: def repeat_last_axis(array, count):
      
      return as_strided(array, array.shape + (count,), array.strides + (0,))
docs: Restride `array` to repeat `count` times along the last axis.
  
      Parameters
      ----------
      array : np.array
          The array to restride.
      count : int
          Number of times to repeat `array`.
  
      Returns
      -------
      result : array
          Array of shape array.shape + (count,) composed of `array` repeated
          `count` times along the last axis.
  
      Example
      -------
      >>> from numpy import arange
      >>> a = arange(3); a
      array([0, 1, 2])
      >>> repeat_last_axis(a, 2)
      array([[0, 0],
             [1, 1],
             [2, 2]])
      >>> repeat_last_axis(a, 4)
      array([[0, 0, 0, 0],
             [1, 1, 1, 1],
             [2, 2, 2, 2]])
  
      Notes
      ----
      The resulting array will share memory with `array`.  If you need to assign
      to the input or output, you should probably make a copy first.
  
      See Also
      --------
      repeat_last_axis