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

add some non-differentiable functions

Browse Source
This commit is contained in:
Mike J Innes 2017-09-01 23:33:05 -04:00
parent 387686eb41
commit 107d9daa8f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tracker/lib.jl
View File

@ -27,6 +27,9 @@ Base.sum(xs::TrackedScalar, dim...) = xs
back!(::typeof(sum), Δ, xs::TrackedArray, dim...) = back!(xs, similar(xs.x) .= Δ) back!(::typeof(sum), Δ, xs::TrackedArray, dim...) = back!(xs, similar(xs.x) .= Δ)
Base.maximum(xs::TrackedArray, args...) = maximum(xs.x, args...)
Base.findfirst(xs::TrackedArray, args...) = findfirst(xs.x, args...)
# BLAS # BLAS
a::TrackedMatrix * b::TrackedMatrix = TrackedArray(Call(*, a, b)) a::TrackedMatrix * b::TrackedMatrix = TrackedArray(Call(*, a, b))