Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Update visualize.jl #18515

Merged
merged 4 commits into from
Jun 16, 2020
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion julia/src/visualize.jl
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,8 @@ function _format_graphviz_edge(io::IOBuffer, head, tail, attrs)
_format_graphviz_attr(io, attrs)
end
function _extract_shape(str :: AbstractString)
shape = matchall(r"\d+", str)
shape = collect(m.match for m in eachmatch(r"\d+", str))

shape = reverse(shape) # JSON in libmxnet has reversed shape (column vs row majoring)
return "(" * join(shape, ",") * ")"
end