예제 #1
0
 def test_poly_shapes(self):
     for shape in [(3, ), (3, 4), (3, 4, 2)]:
         input_dim = shape[-1]
         self._test_shape(shape, T.polynomial(input_dim))
예제 #2
0
 def test_poly_jacobians(self):
     for input_dim in [2, 5, 10]:
         self._test_jacobian(input_dim, T.polynomial(input_dim))