general vec/mat multiply grad

FluxML · Jul 20, 2019 · db7e04c · db7e04c
1 parent acefe2f
commit db7e04c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lib/array.jl b/src/lib/array.jl
@@ -412,7 +412,7 @@ Base.:*(x::TrackedVector,y::Adjoint{T,<:AbstractVector{T}}) where {T} = track(*,
 Base.:*(x::Adjoint{T,<:AbstractMatrix{T}},y::TrackedVector) where {T} = track(*, x, y)
 Base.:*(x::TrackedVector,y::Adjoint{T,<:AbstractMatrix{T}}) where {T} = track(*, x, y)
 
-@grad a::AbstractMatrix * b::AbstractVecOrMat =
+@grad a::AbstractVecOrMat * b::AbstractVecOrMat =
   data(a)*data(b), Δ -> (Δ * transpose(b), transpose(a) * Δ)
 
 # NNlib