Merge pull request #3 from devmotion/similar

Fix FluxML/Flux.jl#689
FluxML · Mar 25, 2019 · 38e3a88 · 38e3a88
2 parents ceba6f4 + 0372525
commit 38e3a88
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/lib/array.jl b/src/lib/array.jl
@@ -85,10 +85,7 @@ end
 Base.size(x::TrackedArray, i::Integer, j::Integer, is::Integer...) =
   size(data(x), i, j, is...)
 
-Base.similar(x::TrackedArray, dims::Union{AbstractUnitRange,Integer}...) =
-  similar(data(x), dims...)
-
-Base.similar(x::TrackedArray, T::Type) = similar(data(x), T)
+Base.similar(x::TrackedArray, T::Type, dims::Dims) = similar(data(x), T, dims)
 
 for op in [:(==), :≈]
     @eval Base.$op(x::TrackedArray, y::AbstractArray) = Base.$op(data(x), y)