brainpy.math.surrogate.ReluGrad#

class brainpy.math.surrogate.ReluGrad(alpha=0.3, width=1.0)[source]#

Judge spiking state with the ReLU gradient function.

See also

relu_grad

__init__(alpha=0.3, width=1.0)[source]#

Methods

__init__([alpha, width])

call(x)

Call the function for surrogate gradient propagation.

surrogate_fun(x)

The surrogate function.

surrogate_grad(dz, x)

The gradient for the surrogate function.

true_fun(x)

The original true function.