Threshold
Threshold(threshold, value)
Keeps every element above threshold and replaces the rest with value.
The module form of clika_runtime.nn.functional.threshold: construct it
once with the settings, then call it like a function.
Args: threshold: the comparison bound. value: the replacement for elements at or below the bound.
__init__
__init__(self, threshold: 'float', value: 'float') -> 'None'
Initialize self. See help(type(self)) for accurate signature.
extra_repr
extra_repr(self) -> 'str'
extra_repr() -> str
One line of per-class detail for :meth:__repr__; a layer prints
its geometry here (in_features=64, out_features=256).
forward
forward(self, input: 'Tensor') -> 'Tensor'
forward(input) -> Tensor
Runs clika_runtime.nn.functional.threshold on input with the
stored settings; returns a new tensor.