JaxGraphShape.tree_unflatten
============================

.. currentmodule:: energnn.graph

.. automethod:: JaxGraphShape.tree_unflatten