JaxGraph.tree_unflatten
=======================

.. currentmodule:: energnn.graph

.. automethod:: JaxGraph.tree_unflatten