JaxHyperEdgeSet.to_numpy_hyper_edge_set

JaxHyperEdgeSet.to_numpy_hyper_edge_set()[source]

Convert a jax.numpy hyper-edge set for GNN processing to a classical numpy hyper-edge set.

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

Returns:

A classical HyperEdgeSet object with NumPy arrays.

Return type:

HyperEdgeSet