Update src/gather.jl

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