jax.lax.broadcast#

jax.lax.broadcast(operand, sizes)[source]#

Broadcasts an array, adding new leading dimensions

Parameters
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.