def check_backward(self, x_data, y_grad, axis=None, keepdims=False):
     gradient_check.check_backward(functions.Max(axis, keepdims),
                                   x_data,
                                   y_grad,
                                   eps=1e-5,
                                   rtol=1e-3,
                                   atol=1e-3)
Exemple #2
0
 def test_duplicate_axis(self):
     with self.assertRaises(ValueError):
         functions.Max((0, 0))
Exemple #3
0
 def test_invalid_axis_type_in_tuple(self):
     with self.assertRaises(TypeError):
         functions.Max((1, 'x'))
Exemple #4
0
 def test_invalid_axis_type(self):
     with self.assertRaises(TypeError):
         functions.Max([0])