LaTeX TikZ 绘制神经网络图

LaTeX TikZ 绘制神经网络图

其中，\tikzstyle{inputNode}=[circle,draw=blue!50,fill=blue!10,minimum size=10pt,inner sep=0pt]为神经元形状、颜色、填充、大小等信息，\node[inputNode, thick] (i1) at (0, 0.5) {};表示神经元的坐标及名称，\draw[stateTransition] (i1) -- (h11);为绘制从神经元i1指向神经元h11的箭头，\draw[stateTransition] (-1, 0.5) -- node[above] {$x_1$} (i1);为绘制输入箭头，\draw[stateTransition] (o) -- node[above] {$\hat{y}$} (9, 0);为绘制输出箭头，\node[above=of h21, align=center] (l2) {Hidden \\ layer};为标注神经网络的部分。

%!TEX program = pdflatex
\documentclass[crop, tikz]{standalone}
\usepackage{tikz}

\usetikzlibrary{positioning}

\tikzstyle{inputNode}=[circle,draw=blue!50,fill=blue!10,minimum size=10pt,inner sep=0pt]
\tikzstyle{stateTransition}=[-stealth, thick]

\begin{document}
\begin{tikzpicture}

    \node[inputNode, thick] (i1) at (0, 0.5) {};
    \node[inputNode, thick] (i2) at (0, 0) {};
    \node[inputNode, thick] (i3) at (0, -0.5) {};
	
    \node[inputNode, thick] (h11) at (2, 2.25) {};
    \node[inputNode, thick] (h12) at (2, 1.75) {};
    \node[inputNode, thick] (h13) at (2, 1.25) {};
    \node[inputNode, thick] (h14) at (2, 0.75) {};
    \node[inputNode, thick] (h15) at (2, 0.25) {};
    \node[inputNode, thick] (h16) at (2, -0.25) {};
    \node[inputNode, thick] (h17) at (2, -0.75) {};
    \node[inputNode, thick] (h18) at (2, -1.25) {};
    \node[inputNode, thick] (h19) at (2, -1.75) {};
    \node[inputNode, thick] (h110) at (2, -2.25) {};
    
    \node[inputNode, thick] (h21) at (4, 1.5) {};
    \node[inputNode, thick] (h22) at (4, 1) {};
    \node[inputNode, thick] (h23) at (4, 0.5) {};
    \node[inputNode, thick] (h24) at (4, 0) {};
    \node[inputNode, thick] (h25) at (4, -0.5) {};
    \node[inputNode, thick] (h26) at (4, -1) {};
    \node[inputNode, thick] (h27) at (4, -1.5) {};

    \node[inputNode, thick] (h31) at (6, 1) {};
    \node[inputNode, thick] (h32) at (6, 0.5) {};
    \node[inputNode, thick] (h33) at (6, 0) {};
    \node[inputNode, thick] (h34) at (6, -0.5) {};
    \node[inputNode, thick] (h35) at (6, -1) {};

	\node[inputNode, thick] (o) at (8, 0) {};


    \draw[stateTransition] (i1) -- (h11);
    \draw[stateTransition] (i1) -- (h12);
    \draw[stateTransition] (i1) -- (h13);
    \draw[stateTransition] (i1) -- (h14);
    \draw[stateTransition] (i1) -- (h15);
    \draw[stateTransition] (i1) -- (h16);
    \draw[stateTransition] (i1) -- (h17);
    \draw[stateTransition] (i1) -- (h18);
    \draw[stateTransition] (i1) -- (h19);
    \draw[stateTransition] (i1) -- (h110);

    \draw[stateTransition] (i2) -- (h11);
    \draw[stateTransition] (i2) -- (h12);
    \draw[stateTransition] (i2) -- (h13);
    \draw[stateTransition] (i2) -- (h14);
    \draw[stateTransition] (i2) -- (h15);
    \draw[stateTransition] (i2) -- (h16);
    \draw[stateTransition] (i2) -- (h17);
    \draw[stateTransition] (i2) -- (h18);
    \draw[stateTransition] (i2) -- (h19);
    \draw[stateTransition] (i2) -- (h110);

    \draw[stateTransition] (i3) -- (h11);
    \draw[stateTransition] (i3) -- (h12);
    \draw[stateTransition] (i3) -- (h13);
    \draw[stateTransition] (i3) -- (h14);
    \draw[stateTransition] (i3) -- (h15);
    \draw[stateTransition] (i3) -- (h16);
    \draw[stateTransition] (i3) -- (h17);
    \draw[stateTransition] (i3) -- (h18);
    \draw[stateTransition] (i3) -- (h19);
    \draw[stateTransition] (i3) -- (h110);

    \draw[stateTransition] (h11) -- (h21);
    \draw[stateTransition] (h11) -- (h22);
    \draw[stateTransition] (h11) -- (h23);
    \draw[stateTransition] (h11) -- (h24);
    \draw[stateTransition] (h11) -- (h25);
    \draw[stateTransition] (h11) -- (h26);
    \draw[stateTransition] (h11) -- (h27);

    \draw[stateTransition] (h12) -- (h21);
    \draw[stateTransition] (h12) -- (h22);
    \draw[stateTransition] (h12) -- (h23);
    \draw[stateTransition] (h12) -- (h24);
    \draw[stateTransition] (h12) -- (h25);
    \draw[stateTransition] (h12) -- (h26);
    \draw[stateTransition] (h12) -- (h27);

    \draw[stateTransition] (h13) -- (h21);
    \draw[stateTransition] (h13) -- (h22);
    \draw[stateTransition] (h13) -- (h23);
    \draw[stateTransition] (h13) -- (h24);
    \draw[stateTransition] (h13) -- (h25);
    \draw[stateTransition] (h13) -- (h26);
    \draw[stateTransition] (h13) -- (h27);

    \draw[stateTransition] (h14) -- (h21);
    \draw[stateTransition] (h14) -- (h22);
    \draw[stateTransition] (h14) -- (h23);
    \draw[stateTransition] (h14) -- (h24);
    \draw[stateTransition] (h14) -- (h25);
    \draw[stateTransition] (h14) -- (h26);
    \draw[stateTransition] (h14) -- (h27);

    \draw[stateTransition] (h15) -- (h21);
    \draw[stateTransition] (h15) -- (h22);
    \draw[stateTransition] (h15) -- (h23);
    \draw[stateTransition] (h15) -- (h24);
    \draw[stateTransition] (h15) -- (h25);
    \draw[stateTransition] (h15) -- (h26);
    \draw[stateTransition] (h15) -- (h27);

    \draw[stateTransition] (h16) -- (h21);
    \draw[stateTransition] (h16) -- (h22);
    \draw[stateTransition] (h16) -- (h23);
    \draw[stateTransition] (h16) -- (h24);
    \draw[stateTransition] (h16) -- (h25);
    \draw[stateTransition] (h16) -- (h26);
    \draw[stateTransition] (h16) -- (h27);

    \draw[stateTransition] (h17) -- (h21);
    \draw[stateTransition] (h17) -- (h22);
    \draw[stateTransition] (h17) -- (h23);
    \draw[stateTransition] (h17) -- (h24);
    \draw[stateTransition] (h17) -- (h25);
    \draw[stateTransition] (h17) -- (h26);
    \draw[stateTransition] (h17) -- (h27);

    \draw[stateTransition] (h18) -- (h21);
    \draw[stateTransition] (h18) -- (h22);
    \draw[stateTransition] (h18) -- (h23);
    \draw[stateTransition] (h18) -- (h24);
    \draw[stateTransition] (h18) -- (h25);
    \draw[stateTransition] (h18) -- (h26);
    \draw[stateTransition] (h18) -- (h27);

    \draw[stateTransition] (h19) -- (h21);
    \draw[stateTransition] (h19) -- (h22);
    \draw[stateTransition] (h19) -- (h23);
    \draw[stateTransition] (h19) -- (h24);
    \draw[stateTransition] (h19) -- (h25);
    \draw[stateTransition] (h19) -- (h26);
    \draw[stateTransition] (h19) -- (h27);

    \draw[stateTransition] (h110) -- (h21);
    \draw[stateTransition] (h110) -- (h22);
    \draw[stateTransition] (h110) -- (h23);
    \draw[stateTransition] (h110) -- (h24);
    \draw[stateTransition] (h110) -- (h25);
    \draw[stateTransition] (h110) -- (h26);
    \draw[stateTransition] (h110) -- (h27);

    \draw[stateTransition] (h21) -- (h31);
    \draw[stateTransition] (h21) -- (h32);
    \draw[stateTransition] (h21) -- (h33);
    \draw[stateTransition] (h21) -- (h34);
    \draw[stateTransition] (h21) -- (h35);

    \draw[stateTransition] (h22) -- (h31);
    \draw[stateTransition] (h22) -- (h32);
    \draw[stateTransition] (h22) -- (h33);
    \draw[stateTransition] (h22) -- (h34);
    \draw[stateTransition] (h22) -- (h35);

    \draw[stateTransition] (h23) -- (h31);
    \draw[stateTransition] (h23) -- (h32);
    \draw[stateTransition] (h23) -- (h33);
    \draw[stateTransition] (h23) -- (h34);
    \draw[stateTransition] (h23) -- (h35);

    \draw[stateTransition] (h24) -- (h31);
    \draw[stateTransition] (h24) -- (h32);
    \draw[stateTransition] (h24) -- (h33);
    \draw[stateTransition] (h24) -- (h34);
    \draw[stateTransition] (h24) -- (h35);

    \draw[stateTransition] (h25) -- (h31);
    \draw[stateTransition] (h25) -- (h32);
    \draw[stateTransition] (h25) -- (h33);
    \draw[stateTransition] (h25) -- (h34);
    \draw[stateTransition] (h25) -- (h35);

    \draw[stateTransition] (h26) -- (h31);
    \draw[stateTransition] (h26) -- (h32);
    \draw[stateTransition] (h26) -- (h33);
    \draw[stateTransition] (h26) -- (h34);
    \draw[stateTransition] (h26) -- (h35);

    \draw[stateTransition] (h27) -- (h31);
    \draw[stateTransition] (h27) -- (h32);
    \draw[stateTransition] (h27) -- (h33);
    \draw[stateTransition] (h27) -- (h34);
    \draw[stateTransition] (h27) -- (h35);

    \draw[stateTransition] (h31) -- (o);
    \draw[stateTransition] (h32) -- (o);
    \draw[stateTransition] (h33) -- (o);
    \draw[stateTransition] (h34) -- (o);
    \draw[stateTransition] (h35) -- (o);

    \draw[stateTransition] (-1, 0.5) -- node[above] {$x_1$} (i1);
    \draw[stateTransition] (-1, 0) -- node[above] {$x_2$} (i2);
    \draw[stateTransition] (-1, -0.5) -- node[above] {$x_3$} (i3);

    \draw[stateTransition] (o) -- node[above] {$\hat{y}$} (9, 0);

    \node[above=of h21, align=center] (l2) {Hidden \\ layer};
    \node[left=7.9em of l2, align=center] (l1) {Input \\ layer};
    \node[right=7.9em of l2, align=center] (l3) {Output \\ layer};

\end{tikzpicture}
\end{document}

Reference

TikZ deep leaning
一个 TiKZ 神经网络的绘制样例