jax.numpy.printoptions#
- jax.numpy.printoptions(*args, **kwargs)[source]#
Context manager for setting print options.
Set print options for the scope of the with block, and restore the old options at the end. See set_printoptions for the full description of available options.
Examples
>>> import numpy as np
>>> from numpy.testing import assert_equal >>> with np.printoptions(precision=2): ... np.array([2.0]) / 3 array([0.67])
The as-clause of the with-statement gives the current print options:
>>> with np.printoptions(precision=2) as opts: ... assert_equal(opts, np.get_printoptions())
See also