視覺化

flax.nnx.display(*args)

使用 Treescope 美化列印器顯示給定的物件。

如果未安裝 treescope 或程式碼未在 IPython 中執行，則 display 將會改為列印物件。