conversions for dual numbers

FluxML · Jan 22, 2019 · 152ce4a · 152ce4a
1 parent 496dbfa
commit 152ce4a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/tracker/lib/real.jl b/src/tracker/lib/real.jl
@@ -99,6 +99,12 @@ import Base:^
 
 ^(a::TrackedReal, b::Integer) = track(^, a, b)
 
+# Hack for conversions
+
+using ForwardDiff: Dual
+
+(T::Type{<:Real})(x::Dual) = Dual(T(x.value), map(T, x.partials.values))
+
 # Tuples
 
 struct TrackedTuple{T<:Tuple}

diff --git a/test/tracker.jl b/test/tracker.jl
@@ -189,6 +189,8 @@ end
 @test gradtest(x -> meanpool(x, (2,2)), rand(10, 10, 3, 2))
 @test gradtest(x -> meanpool(x, (2,2,2)), rand(5, 5, 5, 3, 2))
 
+@test gradtest(x -> Float64.(x), 5)
+
 @testset "equality & order" begin
     # TrackedReal
     @test param(2)^2 == param(4)