mindspore.numpy.repeat
- mindspore.numpy.repeat(a, repeats, axis=None)[source]
Repeats elements of an array.
- Parameters
a (Tensor) – Input array.
repeats (int or sequence of ints) – The number of repetitions for each element. repeats is broadcasted to fit the shape of the given axis.
axis (int, optional) – The axis along which to repeat values. By default, use the flattened input array, and return a flat output array. Defaults to None.
- Returns
Tensor, output array which has the same shape as a, except along the given axis.
- Raises
ValueError – If axis is out of range.
TypeError – If input a is not a Tensor.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore.numpy as np >>> output = np.repeat(np.array(3), 4) >>> print(output) [3 3 3 3] >>> x = np.array([[1,2],[3,4]]) >>> output = np.repeat(x, 2) >>> print(output) [1 1 2 2 3 3 4 4] >>> output = np.repeat(x, 3, axis=1) >>> print(output) [[1 1 1 2 2 2] [3 3 3 4 4 4]] >>> output = np.repeat(x, [1, 2], axis=0) >>> print(output) [[1 2] [3 4] [3 4]]