jax.numpy.printoptions#

jax.numpy.printoptions(*args,**kwargs)[source]#

Alias ofnumpy.printoptions().

JAX arrays are printed via NumPy, so NumPy’sprintoptionsconfigurations will apply to printed JAX arrays.

See thenumpy.set_printoptions() documentation for detailson the available options and their meanings.