jaxsnn.event.to_nir_data.NIRGraphData

class jaxsnn.event.to_nir_data.NIRGraphData(nodes: Dict[str, Union[nir.data_ir.graph.NIRGraphData, nir.data_ir.graph.NIRNodeData]])

Bases: object

Dictionary of NIRNodeData where each entry represents a NIRNode of a corresponding NIRGraph with its observables.

nodesDict[str, Union[NIRGraphData, NIRNodeData]]

Dictionary of NIRNodeData or NIRGraphData for a NIRGraph.

__init__(nodes: Dict[str, Union[nir.data_ir.graph.NIRGraphData, nir.data_ir.graph.NIRNodeData]]) → None

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(nodes)	Initialize self.
check_nodes(graph)	Check if the nodes in NIRData are a subset of the nodes in the NIRGraph

check_nodes(graph: nir.ir.graph.NIRGraph)

Check if the nodes in NIRData are a subset of the nodes in the NIRGraph

nodes: Dict[str, Union[nir.data_ir.graph.NIRGraphData, nir.data_ir.graph.NIRNodeData]]