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