test_minmax.py 文件源码

def check_forward(self, x_data, axis=None, keepdims=False):
        x = chainer.Variable(x_data)
        y = functions.max(x, axis=axis, keepdims=keepdims)
        self.assertEqual(y.data.dtype, numpy.float32)
        y_expect = self.x.max(axis=axis, keepdims=keepdims)
        self.assertEqual(y.data.shape, y_expect.shape)
        gradient_check.assert_allclose(y_expect, y.data)