dense-block.tex 1.4 KB

123456789101112131415161718192021222324252627282930313233343536
  1. \documentclass{standalone}
  2. \usepackage{amssymb}
  3. \usepackage{tikz}
  4. \usetikzlibrary{arrows.meta}
  5. \usetikzlibrary{shapes.geometric,arrows,positioning}
  6. \begin{document}
  7. \begin{tikzpicture}
  8. \tikzstyle{conv-layer}=[draw,minimum width=3cm,minimum height=1cm]
  9. \tikzstyle{arrow}=[->, -Latex, thick]
  10. \draw[fill=black!10,dotted] (-2.5, -7.0) rectangle (2.5, 1.2);
  11. \node[rectangle, fill=black,minimum width=3cm] (input) at (0, 1) {};
  12. \path[arrow] (0, 2) edge node[right, midway] {$256$-d in} (input.north);
  13. % first path
  14. \node (conv1) at (0,-1.0) [conv-layer] {$k$ @ $3 \times 3$};
  15. \node (conc1) at (0,-2.7) [conv-layer, dashed] {concatenate};
  16. \node (conv2) at (0,-4.4) [conv-layer] {$k$ @ $3 \times 3$};
  17. \node (conc2) at (0,-6.1) [conv-layer, dashed] {concatenate};
  18. \path[arrow] (input.south) edge node[right, midway] {$256$-d in} (conv1.north);
  19. \path[arrow] (conv1.south) edge node[right, midway] {$k$-d in} (conc1.north);
  20. \path[arrow] (conc1.south) edge node[right, midway] {$(256 + k)$-d in} (conv2.north);
  21. \path[arrow] (conv2.south) edge node[right, midway] {$k$-d in} (conc2.north);
  22. \path[arrow] (input.south west) edge[bend right] node[right, midway] {} (conc1.north west);
  23. \path[arrow] (conc1.south west) edge[bend right] node[right, midway] {} (conc2.north west);
  24. \path[arrow] (conc2.south) edge node[right] {$(256+2k)$-d out} (0, -8);
  25. \end{tikzpicture}
  26. \end{document}