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

rm logsigmoid

Browse Source
This commit is contained in:
Mike J Innes 2018-02-13 14:52:29 +00:00
parent 2f29733888
commit 49584fb72b
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tracker/array.jl
View File

@ -199,7 +199,7 @@ end
# NNlib # NNlib
using NNlib using NNlib
import NNlib: softmax, ∇softmax, logsoftmax, ∇logsoftmax, logσ, ∇logσ, conv2d, pool import NNlib: softmax, ∇softmax, logsoftmax, ∇logsoftmax, conv2d, pool
softmax(xs::TrackedArray) = track(softmax, xs) softmax(xs::TrackedArray) = track(softmax, xs)
@ -209,10 +209,6 @@ logsoftmax(xs::TrackedArray) = track(logsoftmax, xs)
back(::typeof(logsoftmax), Δ, xs) = @back(xs, ∇logsoftmax(Δ, data(xs))) back(::typeof(logsoftmax), Δ, xs) = @back(xs, ∇logsoftmax(Δ, data(xs)))
logσ(xs::TrackedArray) = TrackedArray(Call(logσ, xs))
back(::typeof(logσ), Δ, xs) = @back(xs, ∇logσ(Δ, data(xs)))
# TODO: can store kwargs efficiently in namedtuples # TODO: can store kwargs efficiently in namedtuples
_conv2d(x, w, stride, pad) = conv2d(x, w, stride = stride, padding = pad) _conv2d(x, w, stride, pad) = conv2d(x, w, stride = stride, padding = pad)