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

numerical stability

Browse Source
This commit is contained in:
Mike J Innes 2018-02-17 11:56:03 +00:00
parent dda545a24a
commit 11511982a4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tracker/array.jl
View File

@ -160,7 +160,7 @@ Base.std(x::TrackedArray, dim; mean = Base.mean(x, dim)) =
Base.norm(x::TrackedArray, p::Real = 2) =
p == 1 ? sum(abs.(x)) :
p == 2 ? sqrt(sum(abs2.(x))) :
p == 2 ? sqrt(sum(abs2.(x) .+ 1e-6)) :
error("$p-norm not supported")
back(::typeof(mean), Δ, xs::TrackedArray) = back(xs, similar(xs.data) .= Δ ./ length(xs.data))