From 892a779ed1c84c10e80dd18cf382d8b1ca86c25e Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 1 Sep 2017 11:42:18 -0400 Subject: [PATCH] tracked transpose --- src/tracker/lib.jl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/tracker/lib.jl b/src/tracker/lib.jl index c300ff16..a173ccac 100644 --- a/src/tracker/lib.jl +++ b/src/tracker/lib.jl @@ -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))