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

tracked transpose

Browse Source
This commit is contained in:
Mike J Innes 2017-09-01 11:42:18 -04:00
parent b95dae1868
commit 892a779ed1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tracker/lib.jl
View File

@ -16,6 +16,9 @@ Base.:-(xs::TrackedArray) = TrackedArray(Call(-, xs))
back!(::typeof(-), Δ, xs::TrackedArray) = back!(xs, -Δ)
Base.transpose(xs::TrackedArray) = TrackedArray(Call(transpose, xs))
Base.ctranspose(xs::TrackedArray) = TrackedArray(Call(ctranspose, xs))
# Reductions
Base.sum(xs::TrackedArray, dim) = TrackedArray(Call(sum, xs, dim))