(self)
| 71 | self.assertAlmostEqual(neg.mean(), 0) |
| 72 | |
| 73 | def test_tanh(self): |
| 74 | X = np.ones((5, 5)).astype(np.float32) - 0.5 |
| 75 | |
| 76 | workspace.FeedBlob("x", X) |
| 77 | model = ModelHelper(name="test_model") |
| 78 | brew.tanh(model, "x", "out_tanh") |
| 79 | model.Validate() |
| 80 | workspace.RunNetOnce(model.param_init_net) |
| 81 | workspace.RunNetOnce(model.net) |
| 82 | |
| 83 | out = workspace.FetchBlob("out_tanh") |
| 84 | self.assertAlmostEqual(out.mean(), np.tanh(0.5), places=5) |
| 85 | |
| 86 | def test_validate(self): |
| 87 | model = ModelHelper(name="test_model") |