Avoid undef error

Signed-off-by: ErikQQY <2283984853@qq.com>

Author: ErikQQY

src/vector_of_array.jl

@@ -736,7 +736,21 @@ end
 end
 
 @inline function Base.similar(VA::VectorOfArray, dims::N) where {N}
-    VectorOfArray(similar(VA.u, dims))
+    l = length(VA)
+    if dims <= l
+        VectorOfArray([similar(VA[:, i]) for i in 1:l])
+    else
+        VectorOfArray([[similar(VA[:, i]) for i in 1:l]; [similar(VA.u[end]) for _ in (l+1):dims]])
+    end
+end
+
+@inline function Base.similar(VA::VectorOfArray, ::Type{T}, dims::N) where {N, T}
+    l = length(VA)
+    if dims <= l
+        VectorOfArray([similar(VA[:, i], T) for i in 1:l])
+    else
+        VectorOfArray([[similar(VA[:, i], T) for i in 1:l]; [similar(VA.u[end], T) for _ in (l+1):dims]])
+    end
 end
 
 @inline function Base.similar(VA::VectorOfArray{T, N, AT},