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)	Time gridded data of shape (n_samples, n_time_steps, n_neurons) with binary entries (0 or 1) indicating whether a neuron spiked at a particular time step dt.

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)