mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

tweak rank promotion warning docs

Browse Source
This commit is contained in:
Matthew Johnson 2019-08-25 14:36:23 -07:00
parent 94fafe880c
commit c6f430a831
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/rank_promotion_warning.rst
View File

@ -24,6 +24,8 @@ allowed just like regular NumPy. The configuration option is named
:code:`jax_numpy_rank_promotion` and it can take on string values
:code:`allow`, :code:`warn`, and :code:`raise`. The default setting is
:code:`warn`, which raises a warning on the first occurrence of rank promotion.
The :code:`raise` setting raises an error on rank promotion, and :code:`allow`
allows rank promotion without warning or error.
As with most other JAX configuration options, you can set this option in
several ways. One is by using :code:`jax.config` in your code: