norse.torch.functional.threshold

norse.torch.functional.threshold

Functions

circ_dist_fn(x, k)

heavi_circ_fn(x, k)

heavi_erfc_fn(x, k)

heavi_tanh_fn(x, k)

logistic_fn(x, k)

sign(x, method, alpha)

threshold(x, method, alpha)

triangle_fn(x[, alpha])

Classes

CircDist(*args, **kwargs)

Approximation of the heaviside step function as

HeaviCirc(*args, **kwargs)

Approximation of the heaviside step function as

HeaviErfc(*args, **kwargs)

Approximation of the heaviside step function as

HeaviTanh(*args, **kwargs)

Approximation of the heaviside step function as

Logistic(*args, **kwargs)

Probalistic approximation of the heaviside step function as

Triangle(*args, **kwargs)

Triangular/piecewise linear surrogate gradient as in