TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix issue #323

Browse Source
This commit is contained in:
Johnny Chen 2018-08-23 20:58:58 +08:00
parent dfe7578216
commit 7bfe431321
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tracker/array.jl
View File

@ -1,4 +1,4 @@
import Base: *, == import Base: *, ==, ≈
import LinearAlgebra import LinearAlgebra
using Statistics using Statistics
@ -64,6 +64,10 @@ x::TrackedArray == y = data(x) == y
y == x::TrackedArray = y == data(x) y == x::TrackedArray = y == data(x)
x::TrackedArray == y::TrackedArray = data(x) == data(y) x::TrackedArray == y::TrackedArray = data(x) == data(y)
x::TrackedArray ≈ y = data(x) ≈ y
y ≈ x::TrackedArray = y ≈ data(x)
x::TrackedArray ≈ y::TrackedArray = data(x) ≈ data(y)
# Array Stdlib # Array Stdlib
Base.getindex(xs::TrackedArray, i...) = track(getindex, xs, i...) Base.getindex(xs::TrackedArray, i...) = track(getindex, xs, i...)

1
src/tracker/scalar.jl
View File

@ -32,6 +32,7 @@ Base.convert(::Type{TrackedReal{T}}, x::TrackedReal{S}) where {T,S} =
Base.:(<)(x::TrackedReal, y::TrackedReal) = data(x) < data(y) Base.:(<)(x::TrackedReal, y::TrackedReal) = data(x) < data(y)
Base.:(==)(x::TrackedReal, y::TrackedReal) = data(x) == data(y) Base.:(==)(x::TrackedReal, y::TrackedReal) = data(x) == data(y)
Base.:(≈)(x::TrackedReal, y::TrackedReal) = data(x) ≈ data(y)
Base.eps(x::TrackedReal) = eps(data(x)) Base.eps(x::TrackedReal) = eps(data(x))