Gaussian

Contents

Gaussian#

class sinabs.activation.Gaussian(mu: float = 0.0, sigma: float = 0.5, grad_scale: float = 1.0)[source]#

Gaussian surrogate gradient function.

Parameters:

  • mu (float) – The mean of the Gaussian.

  • sigma (float) – The standard deviation of the Gaussian.

  • grad_scale (float) – Scale the gradients arbitrarily.

Contents