jax.lax.broadcast_in_dim

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

Wraps XLA’s BroadcastInDim operator.

Parameters
Return type

Any