add some non-differentiable functions

2017-09-01 23:33:05 -04:00 · 2017-09-01 23:33:05 -04:00 · 107d9daa8f
commit 107d9daa8f
parent 387686eb41
1 changed files with 3 additions and 0 deletions
--- a/src/tracker/lib.jl
+++ b/src/tracker/lib.jl
@ -27,6 +27,9 @@ Base.sum(xs::TrackedScalar, dim...) = xs

 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

 a::TrackedMatrix * b::TrackedMatrix  = TrackedArray(Call(*, a, b))