Sub
Sub(alpha=1.0, activation="identity")
Subtracts other, scaled by alpha, from input elementwise with
broadcasting: input - alpha * other, then the optional fused activation.
The module form of clika_runtime.sub: construct it once with the
settings, then call it like a function.
Args:
alpha: scales other before the subtraction.
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.sub on input, other with the stored
settings; returns a new tensor.