Skip to main content

ClikaRT::ops::index_copy

function

index_copy()

Tensor index_copy(
    Tensor x,
    int64_t dim,
    Tensor indices,
    Tensor src
)

A copy of x with rows of src WRITTEN at indices along dim: out[.., indices[i], ..] = src[.., i, ..]. Prefer unique indices; a duplicated position is written more than once.

Same shape/index-dtype rules as index_add.

Throws

  • ClikaRT::Error: (INVALID_ARGUMENT) when the shapes disagree or the index dtype is wrong (code_name() carries the reason).

Returns: a new tensor shaped like x.

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