code: def repeat_first_axis(array, count):
      
      return as_strided(array, (count,) + array.shape, (0,) + array.strides)
docs: Restride `array` to repeat `count` times along the first axis.
  
      Parameters
      ----------
      array : np.array
          The array to restride.
      count : int
          Number of times to repeat `array`.
  
      Returns
      -------
      result : array
          Array of shape (count,) + array.shape, composed of `array` repeated
          `count` times along the first axis.
  
      Example
      -------
      >>> from numpy import arange
      >>> a = arange(3); a
      array([0, 1, 2])
      >>> repeat_first_axis(a, 2)
      array([[0, 1, 2],
             [0, 1, 2]])
      >>> repeat_first_axis(a, 4)
      array([[0, 1, 2],
             [0, 1, 2],
             [0, 1, 2],
             [0, 1, 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