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,