jax.numpy_rank_promotion#
- jax.numpy_rank_promotion = <jax._src.config.State object>#
Context manager for jax_numpy_rank_promotion config option.
Control NumPy-style automatic rank promotion broadcasting (“allow”, “warn”, or “raise”).
- Parameters:
new_val (Any)