jax.lax.broadcast_in_dim¶

jax.lax.broadcast_in_dim(operand, shape, broadcast_dimensions)[source]¶

Wraps XLA’s BroadcastInDim operator.

Parameters
Return type

Any