parent
0957fa7b3c
commit
3110bf9a9a
@ -0,0 +1,50 @@
|
|||||||
|
import unittest
|
||||||
|
import numpy as np
|
||||||
|
from op_test import OpTest
|
||||||
|
|
||||||
|
|
||||||
|
class TestExp(OpTest):
|
||||||
|
def setUp(self):
|
||||||
|
self.op_type = "exp"
|
||||||
|
self.inputs = {
|
||||||
|
'X': np.random.uniform(0.1, 1, [11, 17]).astype("float32")
|
||||||
|
}
|
||||||
|
self.outputs = {'Y': 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)
|
||||||
|
|
||||||
|
|
||||||
|
class TestRelu(OpTest):
|
||||||
|
def setUp(self):
|
||||||
|
self.op_type = "relu"
|
||||||
|
self.inputs = {'X': np.random.uniform(-1, 1, [4, 4]).astype("float32")}
|
||||||
|
self.outputs = {'Y': np.maximum(self.inputs['X'], 0)}
|
||||||
|
|
||||||
|
def test_check_output(self):
|
||||||
|
self.check_output()
|
||||||
|
|
||||||
|
def test_check_grad(self):
|
||||||
|
self.check_grad(['X'], 'Y', max_relative_error=0.007)
|
||||||
|
|
||||||
|
|
||||||
|
class TestSigmoid(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()
|
@ -1,22 +0,0 @@
|
|||||||
import unittest
|
|
||||||
import numpy as np
|
|
||||||
from op_test import OpTest
|
|
||||||
|
|
||||||
|
|
||||||
class TestExp(OpTest):
|
|
||||||
def setUp(self):
|
|
||||||
self.op_type = "exp"
|
|
||||||
self.inputs = {
|
|
||||||
'X': np.random.uniform(0.1, 1, [11, 17]).astype("float32")
|
|
||||||
}
|
|
||||||
self.outputs = {'Y': 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()
|
|
@ -1,20 +0,0 @@
|
|||||||
import unittest
|
|
||||||
import numpy as np
|
|
||||||
from op_test import OpTest
|
|
||||||
|
|
||||||
|
|
||||||
class TestRelu(OpTest):
|
|
||||||
def setUp(self):
|
|
||||||
self.op_type = "relu"
|
|
||||||
self.inputs = {'X': np.random.uniform(-1, 1, [4, 4]).astype("float32")}
|
|
||||||
self.outputs = {'Y': np.maximum(self.inputs['X'], 0)}
|
|
||||||
|
|
||||||
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()
|
|
@ -1,22 +0,0 @@
|
|||||||
import unittest
|
|
||||||
import numpy as np
|
|
||||||
from op_test import OpTest
|
|
||||||
|
|
||||||
|
|
||||||
class TestSigmoid(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()
|
|
Loading…
Reference in new issue