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

printing

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

13
src/grad/track.jl
View File

@ -40,6 +40,17 @@ function back!(x::Var, Δ)
back!(x.f, Δ) back!(x.f, Δ)
end end
for f in :[Base.size, Base.getindex].args for f in :[Base.size, Base.ndims, Base.similar].args
@eval @inline $f(x::Var, a...) = $f(data(x), a...) @eval @inline $f(x::Var, a...) = $f(data(x), a...)
end end
function Base.showarray(io::IO, X::Var, repr::Bool = true; header = true)
if repr
print(io, "Var(")
Base.showarray(io, data(X), true)
print(io, ")")
else
println(io, summary(X), ":")
Base.showarray(io, data(X), false, header = false)
end
end