Exemplo n.º 1
0
 def test_invalid_axis(self):
     with self.assertRaises(TypeError):
         functions.Rollaxis('a', 0)
Exemplo n.º 2
0
 def test_invalid_start(self):
     with self.assertRaises(TypeError):
         functions.Rollaxis(0, 'a')
Exemplo n.º 3
0
 def check_backward(self, x_data, g_data):
     gradient_check.check_backward(
         functions.Rollaxis(self.axis, self.start), x_data, g_data)