我想使用Flux.jl
和Plots.jl
绘制一个函数及其梯度。
using Flux.Tracker
using Plots
f(x::Float64) = 3x^2 + 2x + 1
df(x::Float64) = Tracker.gradient(f, x)[1]
d2f(x::Float64) = Tracker.gradient(df, x)[1]
plot([f], -2, 2)
plot!([df], -2, 2)
I get:
ERROR: LoadError: MethodError: no method matching Float64(::Flux.Tracker.TrackedReal{Float64})
Closest candidates are:
Float64(::Real, ::RoundingMode) where T<:AbstractFloat at rounding.jl:194
Float64(::T<:Number) where T<:Number at boot.jl:741
Float64(::Int8) at float.jl:60
所以我想的是将
Flux.Tracker.TrackedReal{Float64}
转换为 Float64
。我该怎么做呢?