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))
def test_poly_jacobians(self):
for input_dim in [2, 5, 10]:
self._test_jacobian(input_dim, T.polynomial(input_dim))