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.

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]]