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

number fix

Browse Source
This commit is contained in:
Mike J Innes 2018-02-08 17:04:48 +00:00
parent 0f7a1ec022
commit d1c56ca768
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/tracker/Tracker.jl
View File

@ -28,6 +28,7 @@ mutable struct Tracked{T}
end end
Tracked(f::Call, x) = Tracked{typeof(x)}(f, x) Tracked(f::Call, x) = Tracked{typeof(x)}(f, x)
Tracked(f::Call, x, Δ) = Tracked{typeof(x)}(f, x, Δ)
track(f::Call, x) = Tracked(f, x) track(f::Call, x) = Tracked(f, x)
track(f::Call) = track(f, f()) track(f::Call) = track(f, f())

4
src/tracker/scalar.jl
View File

@ -2,11 +2,11 @@ struct TrackedNumber{T<:Number} <: Number
tracker::Tracked{T} tracker::Tracked{T}
end end
TrackedNumber(x::Number) = TrackedNumber(Tracked(Call(nothing), x)) TrackedNumber(x::Number) = TrackedNumber(Tracked(Call(nothing), x, zero(x)))
tracker(x::TrackedNumber) = x.tracker tracker(x::TrackedNumber) = x.tracker
track(f::Call, x::Number) = TrackedNumber(Tracked(f, x)) track(f::Call, x::Number) = TrackedNumber(Tracked(f, x, zero(x)))
back!(x::TrackedNumber) = back!(x, 1) back!(x::TrackedNumber) = back!(x, 1)