add getindex slice method for vector of numbers

SciML · Feb 27, 2024 · d55dac2 · d55dac2
1 parent 96a09f3
commit d55dac2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -278,6 +278,11 @@ Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
     return Adapt.adapt(__parameterless_type(T),
         reshape(reduce(hcat, vecs), size(A.u[1])..., length(A.u)))
 end
+Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
+    ::NotSymbolic, I::Colon...) where {T <: Number, N}
+    @assert length(I) == ndims(A.u)
+    return A.u[I...]
+end
 
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
     ::NotSymbolic, I::AbstractArray{Bool},