torch.Tensor.scatter_reduce_
Tensor.scatter_reduce_(dim, index, src, reduce, *, include_self=True) -> Tensor
Reduces all values from the "src" tensor to the indices specified
   in the "index" tensor in the "self" tensor using the applied
   reduction defined via the "reduce" argument (""sum"", ""prod"",
   ""mean"", ""amax"", ""amin""). For each value in "src", it is
   reduced to an index in "self" which is specified by its index in
   "src" for "dimension != dim" and by the corresponding value in
   "index" for "dimension = dim". If "include_self="True"", the values
   in the "self" tensor are included in the reduction.
"self", "index" and "src" should all have the same number of
   dimensions. It is also required that "index.size(d) <= src.size(d)"
   for all dimensions "d", and that "index.size(d) <= self.size(d)"
   for all dimensions "d != dim". Note that "index" and "src" do not
   broadcast.
For a 3-D tensor with "reduce="sum"" and "include_self=True" the