Add
Add(alpha=1.0, activation="identity")
Adds ``other``, scaled by alpha, to ``input`` elementwise with
broadcasting: input + alpha * other, then the optional fused activation.
The module form of ``clika_runtime.add``: construct it once with the
settings, then call it like a function.
Args:
alpha: scales ``other`` before the add.
activation: a fused elementwise epilogue on the result; float inputs
only.
__init__
__init__(self, alpha: 'float' = 1.0, activation: '_Activation' = 'identity') -> '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', other: 'Tensor | float | int') -> 'Tensor'
forward(input, other) -> Tensor
Runs clika_runtime.add on input, other with the stored
settings; returns a new tensor.