jax.lax.broadcast

Warning

This page was created from a pull request (#9655).

jax.lax.broadcast¶

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

Broadcasts an array, adding new leading dimensions

Parameters

  • operand (Any) – 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

Any

Returns

An array containing the result.

See also

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

previous

jax.lax.population_count

next

jax.lax.broadcasted_iota