onecold fix
This commit is contained in:
parent
1eee724054
commit
62d780c77f
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user