def test_sylvester_shapes(self): for shape in [(3, ), (3, 4), (3, 4, 2)]: input_dim = shape[-1] self._test_shape(shape, T.sylvester(input_dim))
def test_sylvester_jacobians(self): for input_dim in [2, 5, 10]: self._test_jacobian(input_dim, T.sylvester(input_dim))