From 1115eda6af39cfd87862f991c50e63f3d34586e2 Mon Sep 17 00:00:00 2001 From: pshashk Date: Mon, 20 Aug 2018 16:11:56 +0300 Subject: [PATCH] repeat fix ERROR: UndefVarError: A not defined --- src/tracker/array.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tracker/array.jl b/src/tracker/array.jl index 20568086..fb1e1fff 100644 --- a/src/tracker/array.jl +++ b/src/tracker/array.jl @@ -86,9 +86,9 @@ Base.adjoint(xs::TrackedArray) = track(adjoint, xs) @grad transpose(xs) = transpose(data(xs)), Δ -> (reshape(transpose(Δ), size(xs)),) @grad adjoint(xs) = data(xs)', Δ -> (reshape(Δ', size(xs)),) -Base.repeat(A::TrackedArray; kw...) = track(repeat, A; kw...) +Base.repeat(xs::TrackedArray; kw...) = track(repeat, xs; kw...) -@grad function repeat(xs; inner=ntuple(x->1, ndims(A)), outer=ntuple(x->1, ndims(A))) +@grad function repeat(xs; inner=ntuple(x->1, ndims(xs)), outer=ntuple(x->1, ndims(xs))) repeat(data(xs), inner = inner, outer = outer), function (Δ) Δ′ = zero(xs) S = size(xs)