//clika-runtime/io.clika.runtime/Ops/scatterReduce

scatterReduce

[common]
fun scatterReduce(input: Tensor, dim: Long, index: Tensor, src: Tensor, reduce: ScatterReduceMode, includeSelf: Boolean = true, deterministic: Boolean = false): Tensor

scatterReduce(input: Tensor, dim: Long, index: Tensor, src: Tensor, reduce: ScatterReduceMode, includeSelf: Boolean = true, deterministic: Boolean = false): the scatter_reduce operator. scatter with a REDUCTION at each destination: Sum / Prod / Mean / AMax / AMin (ScatterReduceMode).