jax.numpy.repeatΒΆ
-
jax.numpy.
repeat
(a, repeats, axis=None, *, total_repeat_length=None)[source]ΒΆ Repeat elements of an array.
LAX-backend implementation of
repeat()
. Jax adds the optional total_repeat_length parameter which specifies the total number of repeat, and defaults to sum(repeats). It must be specified for repeat to be compilable. If sum(repeats) is larger than the specified total_repeat_length the remaining values will be discarded. In the case of sum(repeats) being smaller than the specified target length, the final value will be repeated.Original docstring below.
- Parameters
a (array_like) β Input array.
repeats (int or array 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
repeated_array β Output array which has the same shape as a, except along the given axis.
- Return type
See also
tile()
Tile an array.
Examples
>>> np.repeat(3, 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]])