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

付録Bget_item関数の実装（ステップ47の補足）

ステップ47では、DeZeroの関数として、get_item関数の使い方だけを紹介しました。本付録では、その実装について説明します。早速ですが、GetItemクラスとget_item関数のコードを次に示します。

dezero/functions.py

class GetItem(Function):
    def __init__(self, slices):
        self.slices = slices

    def forward(self, x):
        y = x[self.slices]
        return y

    def backward(self, gy):
        x, = self.inputs
        f = GetItemGrad(self.slices, x.shape)
        return f(gy)

def get_item(x, slices):
    return GetItem(slices)(x)

ここで、スライス操作を行う引数を初期化の際にslicesとして受け取ります。そして、forward(x)メソッドでは、単にx[self.slices]として要素の抜き出しを行います。

	DeZeroのforward(x)のxはndarrayインスタンスですが、backward(gy)のgyはVariableインスタンスです。逆伝播の実装では、Variableインスタンスに対して、DeZeroの関数を使って計算を行う必要があります。

また、スライス操作に対応する逆伝播の計算は、DeZeroの関数に存在しません。そのため、別途 ...