Skip to main content

Where

Where()

Elementwise select with broadcasting: ``input`` where ``condition`` holds,
``other`` elsewhere.

The module form of ``clika_runtime.where``: construct it once, then call it
like a function.

__init__​

__init__(self) -> 'None'

Initialize self. See help(type(self)) for accurate signature.

forward​

forward(self, condition: 'Tensor', input: 'Tensor', other: 'Tensor') -> 'Tensor'

forward(condition, input, other) -> Tensor

Runs clika_runtime.where on condition, input, other; returns a new tensor.