hxtorch.spiking.utils.to_nir_data

Translate spikes from a jaxsnn spike representation to NIRGraphData.

Classes

NIRGraphData(nodes, …)	Dictionary of NIRNodeData where each entry represents a NIRNode of a corresponding NIRGraph with its observables.
NIRNodeData(observables, …)	Dictionary of EventData or TimeGriddedData where each entry represents an observable (e.g., spikes, voltages) of a corresponding NIRNode
TimeGriddedData(data, dt)	Either boolean entries indicate whether a binary event is present at a particular time step, or a real-valued signal provides the measurement of a quantity (e.g.

Functions

hxtorch.spiking.utils.to_nir_data.to_nir_data(hxtorch_dict: dict, hxtorch_model) → nir.data_ir.graph.NIRGraphData

hxtorch data comes as torch tensor of size (n_time_steps, batch_size, n_neurons)