Update src/scatter.jl

Co-authored-by: Carlo Lucibello <[email protected]>
FluxML · Apr 15, 2021 · b891685 · b891685
1 parent 2e5765f
commit b891685
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scatter.jl b/src/scatter.jl
@@ -67,7 +67,7 @@ function scatter!(op, dst::AbstractArray, src::AbstractArray, idx::AbstractArray
     dims = _check_dims(dst, src, idx)
     colons = Base.ntuple(_->Colon(), dims)
     for k in CartesianIndices(idx)
-        dst_v = _view(dst, colons, idx, k)
+        dst_v = _view(dst, colons, idx[k])
         src_v = _view(src, colons, k)
         dst_v .= (op).(dst_v, src_v)
     end