Skip to main content

ReLU6

ReLU6(inplace=False)

ReLU capped at 6: clamp(input, 0, 6).

The module form of ``clika_runtime.nn.functional.relu6``: construct it once
with the settings, then call it like a function.

Args:
    inplace: whether the forward writes the result into its input and
        returns that input.

__init__​

__init__(self, inplace: 'bool' = False) -> '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.relu6 on input; returns a new tensor, or input itself, holding the result, when inplace is set.