torch.Tensor.Tensor.scatter_reduce
Tensor.scatter_reduce(dim: number, index: Tensor, src: Tensor, reduce: 'sum' | 'prod' | 'mean' | 'amax' | 'amin', options?: ScatterReduceOptions): Tensor<S, D, Dev>Tensor.scatter_reduce(dim: number, index: Tensor, src: Tensor, reduce: 'sum' | 'prod' | 'mean' | 'amax' | 'amin', includeSelfOrOptions?: boolean | ScatterReduceOptions, options?: ScatterReduceOptions): Tensor<S, D, Dev>Reduces src values into self at indices specified by index along dim using reduce operation. Supported reduce operations: 'sum', 'prod', 'mean', 'amax', 'amin'
Parameters
dimnumberindexTensorsrcTensorreduce'sum' | 'prod' | 'mean' | 'amax' | 'amin'optionsScatterReduceOptionsoptional
Returns
Tensor<S, D, Dev>