norse.torch.utils.plot.plot_spikes_2d

norse.torch.utils.plot.plot_spikes_2d(spikes: torch.Tensor, axes: Optional[matplotlib.axes._axes.Axes] = None, **kwargs)

Plots a 2D diagram of spikes. Works similar to the plot_heatmap_2d() but in black and white.

Example:

>>> import torch
>>> from norse.torch import LIF
>>> from norse.torch.utils.plot import plot_spikes_2d
>>> spikes, _ = LIF()(torch.randn(200, 10))
>>> plot_spikes_2d(spikes)
>>> plt.show()

Arguments:

spikes (torch.Tensor): A tensor of spikes from a single layer in two dimensions. axes (matplotlib.axes.Axes): The matplotlib axis to plot on, if any.

Defaults to matplotlib.pyplot.gca()

kwargs: Specific key-value arguments to style the figure

fed to the matplotlib.pyplot.imshow() function.

Returns:

An matplotlib.axes.Axes object