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

tracker fix

Browse Source
This commit is contained in:
Mike J Innes 2018-02-13 16:15:36 +00:00
parent 1baa7227e3
commit 5ea0ef6764
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/onehot.jl
View File

@ -61,5 +61,5 @@ argmax(y::AbstractMatrix, l...) =
# Ambiguity hack
a::TrackedMatrix * b::OneHotVector = TrackedArray(Tracker.Call(*, a, b))
a::TrackedMatrix * b::OneHotMatrix = TrackedArray(Tracker.Call(*, a, b))
a::TrackedMatrix * b::OneHotVector = Tracker.track(*, a, b)
a::TrackedMatrix * b::OneHotMatrix = Tracker.track(*, a, b)