Skip to main content

ClikaRT::ops::scatter_reduce

function

scatter_reduce()​

Tensor scatter_reduce(
    Tensor x,
    int64_t dim,
    Tensor index,
    Tensor src,
    ScatterReduceMode reduce,
    bool include_self = true,
    bool deterministic = false
)

scatter with a REDUCTION at each destination: Sum / Prod / Mean / AMax / AMin (ScatterReduceMode).

Parameters

  • reduce: the reduction applied where writes collide.
  • include_self: true (default) folds the destination's own starting value into the reduction; false reduces the scattered values alone.
  • deterministic: as scatter_add. Default false.

Throws

  • ClikaRT::Error: (INVALID_ARGUMENT) when dim is out of range, 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 2595