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

ステップ16複雑な計算グラフ（実装編）

本ステップは、前ステップの実装編です。まずは、順伝播時に「世代」を設定する実装を行います。そして、逆伝播の際には、その「世代」の数が大きい順に関数を取り出すようにします。それによって、どのような複雑な計算グラフでも正しい順番で逆伝播が行われます。

16.1　世代の追加

はじめに、VariableクラスとFunctionクラスに、インスタンス変数のgenerationを追加します。このgenerationは、どの“世代”の関数（もしくは変数）であるかを示します。それでは、はじめにVariableクラスのコードから示します。Variableクラスの追加コードは次のとおりです。

steps/step16.py

class Variable:
    def __init__(self, data):
        if data is not None:
            if not isinstance(data, np.ndarray):
                raise TypeError('{} is not supported'.format(type(data)))

        self.data = data
        self.grad = None
        self.creator = None
        self.generation = 0

    def set_creator(self, func):
        self.creator = func
        self.generation = func.generation + 1
    ...

Variableクラスは、generationを0で初期化します。そして、set_creatorメソッドが呼ばれた際に、その親となる関数より1だけ大きい世代となるように設定します。たとえば図16-1 ...