ClikaRT::ops::select

function

select()

Tensor select(
    Tensor x,
    int64_t dim,
    IndexBound index
)

One position along dim (the dim is removed). index is an int literal OR a 0-D integer Tensor (e.g. a shape_host-derived index) so a data-dependent select never reads the value to the host.

Declared in ClikaRT/compute/ops.h, line 3351