allow zeros, ones, infs, nans to accept any array, not just matrices

JuliaLang · Mar 2, 2014 · 517039e · 517039e
1 parent 73d6a45
commit 517039e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/base/array.jl b/base/array.jl
@@ -199,12 +199,12 @@ fill(v, dims::Integer...) = fill!(Array(typeof(v), dims...), v)
 
 for (fname, felt) in ((:zeros,:zero),
                       (:ones,:one),
-                      (:infs,:inf), 
+                      (:infs,:inf),
                       (:nans,:nan))
     @eval begin
-        ($fname){T}(::Type{T}, dims...) = fill!(Array(T, dims...), ($felt)(T))
-        ($fname)(dims...)               = fill!(Array(Float64, dims...), ($felt)(Float64))
-        ($fname){T}(x::AbstractMatrix{T}) = ($fname)(T, size(x, 1), size(x, 2))
+        ($fname){T}(::Type{T}, dims...)  = fill!(Array(T, dims...), ($felt)(T))
+        ($fname)(dims...)                = fill!(Array(Float64, dims...), ($felt)(Float64))
+        ($fname){T}(x::AbstractArray{T}) = ($fname)(T, size(x))
     end
 end