jax.numpy.broadcast_to

jax.numpy.broadcast_to(arr, shape)

Broadcast an array to a new shape.

LAX-backend implementation of broadcast_to().

The JAX version does not necessarily return a view of the input.

Original docstring below.

Parameters

shape (tuple) – The shape of the desired array.

Returns

broadcast – A readonly view on the original array with the given shape. It is typically not contiguous. Furthermore, more than one element of a broadcasted array may refer to a single memory location.

Return type

array