Skip to content

Commit 02501d6

Browse files
committed
Change to broadcast
Signed-off-by: ErikQQY <[email protected]>
1 parent c4e0076 commit 02501d6

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/vector_of_array.jl

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -546,9 +546,7 @@ end
546546

547547
function Base.zero(VA::AbstractVectorOfArray)
548548
val = copy(VA)
549-
for i in eachindex(VA.u)
550-
val.u[i] = zero(VA.u[i])
551-
end
549+
val.u = zero.(VA.u)
552550
return val
553551
end
554552

@@ -732,15 +730,15 @@ function Base.similar(vec::VectorOfArray{
732730
end
733731

734732
@inline function Base.similar(VA::VectorOfArray, ::Type{T} = eltype(VA)) where {T}
735-
VectorOfArray([similar(VA[:, i], T) for i in eachindex(VA.u)])
733+
VectorOfArray(similar.(VA.u, T))
736734
end
737735

738736
@inline function Base.similar(VA::VectorOfArray, dims::N) where {N <: Number}
739737
l = length(VA)
740738
if dims <= l
741-
VectorOfArray([similar(VA[:, i]) for i in 1:l])
739+
VectorOfArray(similar.(VA.u[1:dims]))
742740
else
743-
VectorOfArray([[similar(VA[:, i]) for i in 1:l]; [similar(VA.u[end]) for _ in (l+1):dims]])
741+
VectorOfArray([similar.(VA.u); [similar(VA.u[end]) for _ in (l + 1):dims]])
744742
end
745743
end
746744

0 commit comments

Comments
 (0)