diff --git a/src/rulesets/LinearAlgebra/dense.jl b/src/rulesets/LinearAlgebra/dense.jl index 003054cca..a83a6060e 100644 --- a/src/rulesets/LinearAlgebra/dense.jl +++ b/src/rulesets/LinearAlgebra/dense.jl @@ -399,7 +399,7 @@ end ##### `kron` ##### -function frule((_, Δx, Δy), ::typeof(kron), x, y) +function frule((_, Δx, Δy), ::typeof(kron), x::AbstractVecOrMat{<:Number}, y::AbstractVecOrMat{<:Number}) return kron(x, y), kron(Δx, y) + kron(x, Δy) end