jax.numpy.set_printoptions

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

Alias of numpy.set_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.