jax.numpy.printoptions

jax.numpy.printoptions(*args, **kwargs)

Alias of numpy.printoptions().

JAX arrays are printed via NumPy, so NumPy’s printoptions configurations will apply to printed JAX arrays.

See the numpy.set_printoptions() documentation for details on the available options and their meanings.