Mul
Mul(activation="identity")
Multiplies input by other elementwise with broadcasting, then the
optional fused activation.
The module form of clika_runtime.mul: construct it once with the
settings, then call it like a function.
Args: activation: a fused elementwise epilogue on the result; float inputs only.
__init__
__init__(self, 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.mul on input, other with the stored
settings; returns a new tensor.