jaxsnn.event.from_nir_data.NIRNodeData

class jaxsnn.event.from_nir_data.NIRNodeData(observables: Dict[str, Union[nir.data_ir.graph.EventData, nir.data_ir.graph.TimeGriddedData]])

Bases: object

Dictionary of EventData or TimeGriddedData where each entry represents an observable (e.g., spikes, voltages) of a corresponding NIRNode

observablesDict[str, Union[EventData, TimeGriddedData]]

Dictionary of observables for a NIRNode.

__init__(observables: Dict[str, Union[nir.data_ir.graph.EventData, nir.data_ir.graph.TimeGriddedData]]) → None

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

Methods

__init__(observables)	Initialize self.
check_observables(node)	Check that the shapes of the observables match the node’s output shapes

check_observables(node: nir.ir.node.NIRNode)

Check that the shapes of the observables match the node’s output shapes

observables: Dict[str, Union[nir.data_ir.graph.EventData, nir.data_ir.graph.TimeGriddedData]]