def test_tanh(self): X = np.ones((5, 5)).astype(np.float32) - 0.5 workspace.FeedBlob("x", X) model = ModelHelper(name="test_model") brew.tanh(model, "x", "out_tanh") model.Validate() workspace.RunNetOnce(model.param_init_net) workspace.RunNetOnce(model.net) out = workspace.FetchBlob("out_tanh") self.assertAlmostEqual(out.mean(), 0.46211711)