def _affine_backward(self, x, w, b, dout): layer = Linear(w.shape[0], w.shape[1]) layer.weight = w layer.bias = b tmp = layer.forward(x) layer.backward(dout) return layer.dx, layer.dw, layer.db
def _affine_forward(self, x, w, b): layer = Linear(w.shape[0], w.shape[1]) layer.weight = w layer.bias = b return layer.forward(x)