Fix size of projected index on selectdim

bsc-quantic · Sep 28, 2023 · fc0cd69 · fc0cd69
1 parent 3b15bce
commit fc0cd69
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Slicing.jl b/src/Slicing.jl
@@ -16,8 +16,9 @@ See also: [`view`](@ref).
 function Base.selectdim(path::EinExpr, index::Symbol, i)
     path = deepcopy(path)
 
-    leave = Iterators.filter(∋(index) ∘ head, Leaves(path)) |> first
+    for leave in Iterators.filter(∋(index) ∘ head, Leaves(path))
     leave.size[index] = length(i)
+    end
 
     return path
 end