Ctrl+K

Threshold

Threshold#

class brainpy.dnn.Threshold(threshold, value, inplace=False)[source]#

Thresholds each element of the input Tensor.

Threshold is defined as:

\[\begin{split}y = \begin{cases} x, &\text{ if } x > \text{threshold} \\ \text{value}, &\text{ otherwise } \end{cases}\end{split}\]
Parameters:

  • threshold (float) – The value to threshold at

  • value (float) – The value to replace with

  • inplace (bool) – can optionally do the operation in-place. Default: False

Shape:

  • Input: \((*)\), where \(*\) means any number of dimensions.

  • Output: \((*)\), same shape as the input.

Examples:

>>> import brainpy as bp
>>> import brainpy.math as bm
>>> m = bp.dnn.Threshold(0.1, 20)
>>> input = bm.random.randn(2)
>>> output = m(input)
update(input)[source]#

The function to specify the updating rule.

Return type:

TypeVar(ArrayType, Array, Variable, TrainVar, Array, ndarray)

Contents