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

don't override base method

Browse Source
This commit is contained in:
Mike J Innes 2018-02-16 14:15:40 +00:00
parent fb5b8c7952
commit 60f21d3ff2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tracker/array.jl
View File

@ -113,7 +113,7 @@ back(::typeof(reshape), Δ, xs::TrackedArray, _...) =
back(xs, reshape(Δ, size(xs))) back(xs, reshape(Δ, size(xs)))
function Base.kron(mat1::AbstractMatrix,mat2::AbstractMatrix) function _kron(mat1::AbstractMatrix,mat2::AbstractMatrix)
m1, n1 = size(mat1) m1, n1 = size(mat1)
mat1_rsh = reshape(mat1,(1,m1,1,n1)) mat1_rsh = reshape(mat1,(1,m1,1,n1))
@ -123,6 +123,10 @@ function Base.kron(mat1::AbstractMatrix,mat2::AbstractMatrix)
return reshape(mat1_rsh.*mat2_rsh, (m1*m2,n1*n2)) return reshape(mat1_rsh.*mat2_rsh, (m1*m2,n1*n2))
end end
Base.kron(a::TrackedMatrix, b::TrackedMatrix) = _kron(a, b)
Base.kron(a::TrackedMatrix, b::AbstractMatrix) = _kron(a, b)
Base.kron(a::AbstractMatrix, b::TrackedMatrix) = _kron(a, b)
# Reductions # Reductions
Base.sum(xs::TrackedArray, dim) = track(sum, xs, dim) Base.sum(xs::TrackedArray, dim) = track(sum, xs, dim)