ゼロから作るDeep Learning ❸ ―フレームワーク編

ステップ26計算グラフの可視化（2）

前ステップでは、DOT言語の記述方法を学びました。ここでは、その知識を基に、DeZeroの計算グラフをDOT言語に変換します。具体的には、DeZeroで行った計算をDOT言語へと変換する関数を実装していきます。

26.1　可視化コードの使用例

私たちは、計算グラフを可視化する関数をget_dot_graphという名前で、dezero/utils.pyに実装することにします。まずは、その関数の使用例を先に示します。

import numpy as np
from dezero import Variable
from dezero.utils import get_dot_graph

x0 = Variable(np.array(1.0))
x1 = Variable(np.array(1.0))
y = x0 + x1  # 何らかの計算

# 変数に名前を付ける
x0.name = 'x0'
x1.name = 'x1'
y.name = 'y'

txt = get_dot_graph(y, verbose=False)
print(txt)

# dotファイルとして保存
with open('sample.dot', 'w') as o:
    o.write(txt)

実行結果

digraph g { 4423761088 [label="y", color=orange, style=filled] 4423742632 [label="Add", color=lightblue, style=filled, shape=box] 4403357456 -> 4423742632 4403358016 -> 4423742632 4423742632 ...