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

tree utilities

Browse Source
This commit is contained in:
Mike J Innes 2017-11-07 10:57:27 +00:00
parent 6eb2ec154b
commit 752a9e2808
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/batches/tree.jl
View File

@ -13,6 +13,9 @@ Tree(x::T, xs::Tree{T}...) where T = Tree{T}(x, xs...)
AbstractTrees.children(t::Tree) = t.children AbstractTrees.children(t::Tree) = t.children
AbstractTrees.printnode(io::IO, t::Tree) = show(io, t.value) AbstractTrees.printnode(io::IO, t::Tree) = show(io, t.value)
Base.show(io::IO, t::Type{Tree}) = print(io, "Tree")
Base.show(io::IO, t::Type{Tree{T}}) where T = print(io, "Tree{", T, "}")
function Base.show(io::IO, t::Tree) function Base.show(io::IO, t::Tree)
println(io, typeof(t)) println(io, typeof(t))
print_tree(io, t) print_tree(io, t)
@ -24,3 +27,16 @@ using Juno
render(t) = Juno.Tree(t.value, render.(t.children)) render(t) = Juno.Tree(t.value, render.(t.children))
Juno.Tree(typeof(t), [render(t)]) Juno.Tree(typeof(t), [render(t)])
end end
Base.getindex(t::Tree, i::Integer) = t.children[i]
Base.getindex(t::Tree, i::Integer, is::Integer...) = t[i][is...]
# Utilities
isleaf(t) = isempty(children(t))
leaves(xs::Tree) = map(x -> x.value, Leaves(xs))
Base.map(f, t::Tree, ts::Tree...) =
Tree{Any}(f(map(t -> t.value, (t, ts...))...),
[map(f, chs...) for chs in zip(map(t -> t.children, (t, ts...))...)]...)