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

onecold fix

Browse Source
This commit is contained in:
Mike J Innes 2019-01-24 10:16:41 +00:00
parent 1eee724054
commit 62d780c77f
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/tracker/lib/array.jl
View File

@ -358,6 +358,13 @@ x::TrackedVector * y::TrackedVector = track(*, x, y)
@grad a::AbstractMatrix * b::AbstractVecOrMat = @grad a::AbstractMatrix * b::AbstractVecOrMat =
data(a)*data(b), Δ -> (Δ * transpose(b), transpose(a) * Δ) data(a)*data(b), Δ -> (Δ * transpose(b), transpose(a) * Δ)
# Flux
import ..Flux.onecold
onecold(x::TrackedVector, l...) = onecold(data(x), l...)
onecold(x::TrackedMatrix, l...) = onecold(data(x), l...)
# NNlib # NNlib
using NNlib using NNlib