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.