Heaviside

Heaviside#

:class:~sinabs.activation.Heaviside surrogate gradient.

import matplotlib.pyplot as plt
import torch

import sinabs.activation as sina

x = torch.linspace(-2, 4, 500)
plt.plot(x, sina.Heaviside(window=0.5)(v_mem=x, spike_threshold=1.0), label="Heaviside")
plt.xlabel("Neuron membrane potential")
plt.ylabel("Derivative")
plt.legend()
<matplotlib.legend.Legend at 0x7fb7081f6cd0>
../../_images/a9830798bbd8f5c08f883b6e7d1f1b02b4a469ebd52e9e53eab99947f486ca10.png