separate_hyper_edge_sets

separate_hyper_edge_sets(hyper_edge_set_batch)

Separate a batched HyperEdgeSet into its constituent HyperEdgeSet instances.

The input HyperEdgeSet must have been created by collate_hyper_edge_sets() or otherwise its property “array” must return a 3D array.

Parameters:

hyper_edge_set_batch (HyperEdgeSet) – The batched HyperEdgeSet to unstack.

Returns:

List of HyperEdgeSet instances, each corresponding to one batch element.

Raises:

ValueError – If hyper_edge_set_batch.is_batch is False.

Return type:

list[HyperEdgeSet]