Skip to main content

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.