jax.numpy.broadcast_arrays

jax.numpy.broadcast_arrays(*args)

Broadcast any number of arrays against each other.

LAX-backend implementation of numpy.broadcast_arrays().

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

Original docstring below.

Returns:

broadcasted – These arrays are views on the original arrays. They are typically not contiguous. Furthermore, more than one element of a broadcasted array may refer to a single memory location. If you need to write to the arrays, make copies first. While you can set the writable flag True, writing to a single output value may end up changing more than one location in the output array.

Deprecated since version 1.17: The output is currently marked so that if written to, a deprecation warning will be emitted. A future version will set the writable flag False so writing to it will raise an error.

Return type:

list of arrays

Parameters:

args (jax.typing.ArrayLike)