import unittest import numpy as np from op_test import OpTest class TestSigmoidOp(OpTest): def setUp(self): self.op_type = "sigmoid" self.inputs = { 'X': np.random.uniform(0.1, 1, [11, 17]).astype("float32") } self.outputs = {'Y': 1 / (1 + np.exp(-self.inputs['X']))} def test_check_output(self): self.check_output() def test_check_grad(self): self.check_grad(["X"], "Y", max_relative_error=0.007) if __name__ == '__main__': unittest.main()