define back function right after forward function

This commit is contained in:
Johan Gustafsson 2018-05-02 08:37:30 +02:00
parent 509a2e59f6
commit fb68529169

View File

@ -81,19 +81,6 @@ back(::typeof(ctranspose), Δ, xs) = @back(xs, trim(xs, Δ'))
Base.repmat(x::TrackedVecOrMat, a::Integer...) = track(repmat, x, a...)
Base.repmat(x::TrackedVecOrMat, a::Int64...) = track(repmat, x, a...)
for f in [:vcat, :hcat]
@eval begin
Base.$f(a::TrackedArray...) = track($f, a...)
Base.$f(a::TrackedArray, b::Array...) = track($f, a, b...)
# assumes there is another function to capture Union{Matrix,Vector}... without any TrackedMatrix or TrackedVector
Base.$f(a::Union{TrackedMatrix,TrackedVector,Matrix,Vector}...) = track($f, a...)
end
end
Base.cat(dim::Int, a::TrackedArray...) = track(Base.cat, dim, a...)
Base.cat(dim::Int, a::TrackedArray, b::Array...) = track(Base.cat, dim, a, b...)
function back(::typeof(repmat), Δ, xs::TrackedVecOrMat, m, n=1)
Δ′ = similar(xs.data)
S = size(xs.data)
@ -106,6 +93,16 @@ function back(::typeof(repmat), Δ, xs::TrackedVecOrMat, m, n=1)
back(xs, Δ′)
end
for f in [:vcat, :hcat]
@eval begin
Base.$f(a::TrackedArray...) = track($f, a...)
Base.$f(a::TrackedArray, b::Array...) = track($f, a, b...)
# assumes there is another function to capture Union{Matrix,Vector}... without any TrackedMatrix or TrackedVector
Base.$f(a::Union{TrackedMatrix,TrackedVector,Matrix,Vector}...) = track($f, a...)
end
end
function back(::typeof(vcat), Δ, xs...)
start = 0
for xsi in xs
@ -128,6 +125,9 @@ function back(::typeof(hcat), Δ, xs...)
end
end
Base.cat(dim::Int, a::TrackedArray...) = track(Base.cat, dim, a...)
Base.cat(dim::Int, a::TrackedArray, b::Array...) = track(Base.cat, dim, a, b...)
function back(::typeof(cat), Δ, dim, xs...)
start = 0
for xsi in xs