flopscope.numpy.repeat
fnp.repeat(a, repeats, axis=None)[flopscope source][numpy source]
Repeat each element of an array after themselves
Adapted from NumPy docs np.repeat
Cost
per-operation
Flopscope Context
Repeat elements of an array. Cost: numel(output).
Parameters
- a:array_like
Input array.
- repeats:int or array of ints
The number of repetitions for each element.
repeatsis 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
- repeated_array:ndarray
Output array which has the same shape as
a, except along the given axis.
See also
- we.flops.tile Tile an array.
- we.flops.unique Find the unique elements of an array.
Examples
>>> import flopscope.numpy as fnp
>>> flops.repeat(3, 4)
array([3, 3, 3, 3])
>>> x = flops.array([[1,2],[3,4]])
>>> flops.repeat(x, 2)
array([1, 1, 2, 2, 3, 3, 4, 4])
>>> flops.repeat(x, 3, axis=1)
array([[1, 1, 1, 2, 2, 2],
[3, 3, 3, 4, 4, 4]])
>>> flops.repeat(x, [1, 2], axis=0)
array([[1, 2],
[3, 4],
[3, 4]])