dpnp.repeat

dpnp.repeat(a, repeats, axis=None)[source]

Repeat elements of an array.

For full documentation refer to numpy.repeat.

Parameters:
  • x ({dpnp.ndarray, usm_ndarray}) -- Input array.

  • repeats ({int, tuple, list, range, dpnp.ndarray, usm_ndarray}) -- The number of repetitions for each element. repeats is broadcasted to fit the shape of the given axis. If repeats is an array, it must have an integer data type. Otherwise, repeats must be a Python integer or sequence of Python integers (i.e., a tuple, list, or range).

  • axis ({None, int}, optional) -- The axis along which to repeat values. By default, use the flattened input array, and return a flat output array. Default: None.

Returns:

out -- Output array which has the same shape as a, except along the given axis.

Return type:

dpnp.ndarray

See also

dpnp.tile

Tile an array.

dpnp.unique

Find the unique elements of an array.

Examples

>>> import dpnp as np
>>> x = np.array([3])
>>> np.repeat(x, 4)
array([3, 3, 3, 3])
>>> x = np.array([[1, 2], [3, 4]])
>>> np.repeat(x, 2)
array([1, 1, 2, 2, 3, 3, 4, 4])
>>> np.repeat(x, 3, axis=1)
array([[1, 1, 1, 2, 2, 2],
       [3, 3, 3, 4, 4, 4]])
>>> np.repeat(x, [1, 2], axis=0)
array([[1, 2],
       [3, 4],
       [3, 4]])