Flux.jl/src/Tracker/lib.jl

11 lines
324 B
Julia
Raw Normal View History

2017-08-19 09:14:50 +00:00
import Base: *
2017-08-19 10:11:25 +00:00
a::TrackedMatrix * b::Union{TrackedMatrix,AbstractMatrix} = Var(Call(*, a, b))
a::Union{TrackedMatrix,AbstractMatrix} * b::TrackedMatrix = Var(Call(*, a, b))
2017-08-19 09:14:50 +00:00
function back!(::typeof(*), Δ, a::AbstractArray, b::AbstractArray)
back!(a, A_mul_Bt(Δ, data(b)))
back!(b, At_mul_B(data(a), Δ))
return
end