jax.numpy.broadcast_to#

jax.numpy.broadcast_to(arr, shape)#

Broadcast an array to a new shape.

LAX-backend implementation of numpy.broadcast_to().

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

Original docstring below.

Parameters

shape (tuple or int) – The shape of the desired array. A single integer i is interpreted as (i,).

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