jax.numpy_rank_promotion — JAX documentation (original) (raw)

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)