def test_backward_axis_cpu(self):
for i in range(self.x.ndim):
gy = numpy.ones_like(self.x.max(axis=i)) * self.gy
self.check_backward(self.x, gy, axis=i)
微信
公众号
扫码关注公众号