- jax.experimental.host_callback.id_print(arg, *, result=None, tap_with_device=False, output_stream=None, threshold=None, **kwargs)#
id_tap()with a printing tap function.
Experimental: please give feedback, and expect changes!
On each invocation of the printing tap, the
kwargsif present will be printed first (sorted by keys). Then arg will be printed, with the arrays stringified with
Additional keyword arguments:
tap_with_deviceif True, will print also the device from which the value originates.
output_streamif given then it will be used instead of the built-in
thresholdis passed to