jax.numpy.broadcast_arrays#
- jax.numpy.broadcast_arrays(*args)[source]#
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 (
Union
[Array
,ndarray
,bool_
,number
,bool
,int
,float
,complex
]) –