Skip to main content

ClikaRT::nn::Where

class

Header: ClikaRT/nn/generated/elementwise.h
Inherits: ClikaRT::nn::Module

Elementwise select with broadcasting: input where condition holds, other elsewhere. The module form of ClikaRT::ops::where, which documents the full contract (shapes, dtypes, errors); construct once, then call it like a function.

Member functions​

Where()​

explicit Where(WhereOptions options = {})

Constructs the module from options (every default when omitted).

Declared in ClikaRT/nn/generated/elementwise.h, line 473

forward()​

Tensor forward(
    Tensor condition,
    Tensor input,
    Tensor other
) const

forward_impl, unwrapped: raises ClikaRT::Error on failure.

Declared in ClikaRT/nn/generated/elementwise.h, line 486

operator()()​

Tensor operator()(
    Tensor condition,
    Tensor input,
    Tensor other
) const

Same as forward.

Declared in ClikaRT/nn/generated/elementwise.h, line 491

options()​

const WhereOptions& options() const

The settings this module was constructed with.

Declared in ClikaRT/nn/generated/elementwise.h, line 496