diff --git a/imperative/python/megengine/functional/tensor.py b/imperative/python/megengine/functional/tensor.py index 8ba9596a67589e1aae4b7c3ab15d9e0daa1ec6f8..31053b7835cc0c25173df1a09557b8d425a5763c 100644 --- a/imperative/python/megengine/functional/tensor.py +++ b/imperative/python/megengine/functional/tensor.py @@ -189,7 +189,7 @@ def identity(inp: Tensor) -> Tensor: :return: output tensor. """ op = builtin.Identity() - (data,) = utils.convert_inputs(inp) + (data,) = convert_inputs(inp) (output,) = apply(op, data) return output diff --git a/imperative/python/test/unit/functional/test_tensor.py b/imperative/python/test/unit/functional/test_tensor.py index b7a320b9f4d63a02300ed06daff8afd0325acb18..c17cf310fe45f4a29f10a8b5f833721b3a929b6c 100644 --- a/imperative/python/test/unit/functional/test_tensor.py +++ b/imperative/python/test/unit/functional/test_tensor.py @@ -367,6 +367,12 @@ def test_device(): np.testing.assert_almost_equal(y5.numpy(), y6.numpy()) +def test_identity(): + x = tensor(np.random.random((5, 10)).astype(np.float32)) + y = F.identity(x) + np.testing.assert_equal(y.numpy(), x) + + def copy_test(dst, src): data = np.random.random((2, 3)).astype(np.float32) x = tensor(data, device=src)