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))