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.

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

Parameters

nodes – Dictionary of NIRNodeData or NIRGraphData

Methods

__init__(nodes)	param nodes Dictionary of NIRNodeData or NIRGraphData
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]]