jax.lax.broadcast

jax.lax.broadcast(operand, sizes)

Broadcasts an array, adding new leading dimensions

Parameters

• operand (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – an array

• sizes (Sequence[int]) – a sequence of integers, giving the sizes of new leading dimensions to add to the front of the array.

Return type

Array

Returns

An array containing the result.

See also

jax.lax.broadcast_in_dim : add new dimensions at any location in the array shape.