jax.numpy.tile#
- jax.numpy.tile(A, reps)[source]#
Construct an array by repeating A the number of times given by reps.
LAX-backend implementation of
numpy.tile()
.Original docstring below.
If reps has length
d
, the result will have dimension ofmax(d, A.ndim)
.If
A.ndim < d
, A is promoted to be d-dimensional by prepending new axes. So a shape (3,) array is promoted to (1, 3) for 2-D replication, or shape (1, 1, 3) for 3-D replication. If this is not the desired behavior, promote A to d-dimensions manually before calling this function.If
A.ndim > d
, reps is promoted to A.ndim by pre-pending 1’s to it. Thus for an A of shape (2, 3, 4, 5), a reps of (2, 2) is treated as (1, 1, 2, 2).Note : Although tile may be used for broadcasting, it is strongly recommended to use numpy’s broadcasting operations and functions.
- Parameters
A (array_like) – The input array.
reps (array_like) – The number of repetitions of A along each axis.
- Returns
c – The tiled output array.
- Return type
ndarray