Seonglae Chopip install treescope
import treescope treescope.register_as_default() treescope.active_autovisualizer.set_globally(treescope.ArrayAutovisualizer()) treescope.basic_interactive_setup(autovisualize_arrays=True)
Treescope — treescope
Treescope is an interactive HTML pretty-printer and N-dimensional array
(“tensor”) visualizer, designed for machine learning and neural networks
research in IPython notebooks. It’s a drop-in replacement for the standard
IPython/Colab renderer, and adds support for:
https://treescope.readthedocs.io/en/stable/