def test_squeeze(self):
"""Test for squeeze operator"""
node_def = helper.make_node("Squeeze", ["ip1"], ["op1"], axes=[1, 3])
ip1 = self._random_array([3, 1, 2, 1, 4])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.squeeze(ip1, axis=[1, 3]))
def test_tanh(self):
"""Test for tanh operator"""
node_def = helper.make_node("Tanh", ["ip1"], ["op1"])
ip1 = self._random_array([1, 10])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.tanh(ip1))
def test_abs(self):
"""Test for abs operator"""
node_def = helper.make_node("Abs", ["ip1"], ["ip2"])
ip1 = self._random_array([1, 1000])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.abs(ip1))
def test_sqrt(self):
"""Test for sqrt operator"""
node_def = helper.make_node("Sqrt", ["ip1"], ["op1"])
ip1 = self._random_array([1, 1000])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.sqrt(ip1))
def test_reciprocal(self):
"""Test for reciprocal operator"""
node_def = helper.make_node("Reciprocal", ["ip1"], ["op1"])
ip1 = self._random_array([1, 1000])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.reciprocal(ip1))
def test_neg(self):
"""Test for neg operator"""
node_def = helper.make_node("Neg", ["ip1"], ["op1"])
ip1 = self._random_array([1, 1000])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.negative(ip1))
def test_relu(self):
"""Test for relu operator"""
node_def = helper.make_node("Relu", ["ip1"], ["op1"])
ip1 = self._random_array([1, 256])
output = mxnet_backend.run_node(node_def, [ip1])[0]
npt.assert_almost_equal(output, np.maximum(ip1, 0))
def test_log(self):
"""Test for log operator"""
node_def = helper.make_node("Log", ["input1"], ["output"])
input1 = self._random_array([1, 10])
output = mxnet_backend.run_node(node_def, [input1])[0]
npt.assert_almost_equal(output, np.log(input1))
