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)
