Note
Go to the end to download the full example code
Heaviside#
Heaviside surrogate gradient.
<matplotlib.legend.Legend object at 0x7fab7d08a8d0>
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()
Total running time of the script: (0 minutes 0.131 seconds)