能够绘制神经网络架构的绘图程序

能够绘制神经网络架构的绘图程序

问题

也许这个问题不太适合这个会场。

以下是我在 Andrew Ng 的深度学习课程上找到的神经网络架构。尽管它看起来不像是使用 tikz 绘制的(或者更像任务专用模块,如情节神经网络)。

我想知道是否有软件(包)能够绘制这样的图表。我猜最困难的部分是附加数学符号(即使这个图表也无法完美地做到这一点)。

答案1

这将给你一个开始。箭头是自动绘制的,方框是用 完成的shapes.multipartsplit row希望风格是不言自明的,其余部分完成得非常快。

\documentclass{article}
\usepackage{tikz}
\usetikzlibrary{matrix,shapes.multipart}
\begin{document}
  \begin{tikzpicture}[font=\sffamily,every text node part/.style={text
  width=2cm,align=center},split row/.style={nodes={draw,rectangle split,
    rectangle split draw splits=true,rectangle split parts=#1}}]
    \matrix[matrix of nodes,
    row 2/.style={split row=2},
    row 3/.style={split row=4},
    row sep=2em,column sep=2em,
    execute at end matrix={% automatically add vertical arrows
    \foreach \Y [evaluate=\Y as \Z using {int(\Y-1)}] in 
        {2,...,\the\pgfmatrixcurrentrow}
    {\foreach \X in {1,...,\the\pgfmatrixcurrentcolumn}
    {\draw[-stealth] (\tikzmatrixname-\Y-\X) -- (\tikzmatrixname-\Z-\X);}}}
    ] { $y^1$ & $y^2$ \\
      Sigmoid\nodepart{two}Dense & Sigmoid\nodepart{two}Dense\\     
     Dropout\nodepart{two}blabla\nodepart{three}blabla%
    \nodepart{four}\vphantom{\raisebox{-1.5ex}{\Huge A}}xyz & 
      Dropout\nodepart{two}blabla\nodepart{three}blabla%
    \nodepart{four}\vphantom{\raisebox{-1.5ex}{\Huge A}}xyz\\
    };
  \end{tikzpicture}
\end{document}

在此处输入图片描述

相关内容