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

vector sum

Browse Source
This commit is contained in:
Mike J Innes 2017-08-22 15:12:12 +01:00
parent 97af9db181
commit cd45df1eca
1 changed files with 3 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/Tracker/lib.jl
View File

@ -1,12 +1,7 @@
import Base: * import Base: *
toarray(xs::AbstractArray, ys::AbstractArray) = ys toarray(xs::AbstractArray, ys::AbstractArray) = ys
toarray(xs::AbstractArray, y) = similar(xs, typeof(y), ()) .= y
function toarray(xs::AbstractArray, y)
y′ = similar(xs, typeof(y), ())
y′[] = y
return y′
end
Base.getindex(xs::TrackedArray, i...) = Base.getindex(xs::TrackedArray, i...) =
TrackedArray(Call(getindex, xs, i...), toarray(xs.x, xs.x[i...])) TrackedArray(Call(getindex, xs, i...), toarray(xs.x, xs.x[i...]))
@ -21,13 +16,10 @@ Base.:-(xs::TrackedArray) = TrackedArray(Call(-, xs))
# Reductions # Reductions
Base.sum(xs::TrackedArray, dim) = TrackedArray(Call(sum, xs, dim))
Base.sum(xs::TrackedArray) = TrackedArray(Call(sum, xs), toarray(xs.x, sum(xs.x))) Base.sum(xs::TrackedArray) = TrackedArray(Call(sum, xs), toarray(xs.x, sum(xs.x)))
function back!(::typeof(sum), Δ, xs::TrackedArray) back!(::typeof(sum), Δ, xs::TrackedArray, dim...) = back!(xs, similar(xs.x) .= Δ)
Δ′ = similar(xs.x)
Δ′ .= Δ
back!(xs, Δ′)
end
# BLAS # BLAS