JaxGraph.to_numpy_graph

JaxGraph.to_numpy_graph()[source]

Convert a jax.numpy graph for GNN processing to a classical numpy graph.

This method transforms the internal JAX arrays of the graph back into standard NumPy arrays, enabling compatibility with non-JAX components.

Returns:

A classical Graph object with NumPy arrays.

Return type:

Graph