def test_reshape_sum_backward(self):
gy = np.ones((3, 4))
self.assertEqual(
reshape_sum_backward(gy, (2, 3, 4, 2), axis=(0, 3),
keepdims=False).shape, (1, 3, 4, 1))
gy = np.array(1.0)
self.assertEqual(
reshape_sum_backward(gy, (3, 4), axis=(0, 1),
keepdims=False).shape, (1, 1))
gy = np.array(1.0)
self.assertEqual(
reshape_sum_backward(gy, (3, 4), axis=None, keepdims=False).shape,
())
def backward(self, gy):
gy = utils.reshape_sum_backward(gy, self.x_shape, self.axis,
self.keepdims)
gx = broadcast_to(gy, self.x_shape)
return gx
def backward(self, *gys: Variable) -> tuple[Variable, ...]:
gy, = gys
gy = utils.reshape_sum_backward(
gy, self.x_shape, self.axis, self.keepdims)
gx = broadcast_to(gy, self.x_shape)
return gx,
