From fd5031f8eb2a6d4d119dc0fedd7daeba1364beda Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 28 Feb 2022 17:10:19 +0800 Subject: [PATCH] docs(functional): replace utils testcode with doctest format GitOrigin-RevId: 8618918ff361d0d65b6e3a58c0dcd954e447625e --- imperative/python/megengine/functional/utils.py | 18 ++++-------------- 1 file changed, 4 insertions(+), 14 deletions(-) diff --git a/imperative/python/megengine/functional/utils.py b/imperative/python/megengine/functional/utils.py index fbab3645..dc0e876a 100644 --- a/imperative/python/megengine/functional/utils.py +++ b/imperative/python/megengine/functional/utils.py @@ -35,21 +35,11 @@ def _assert_equal( Examples: - .. testcode:: + >>> x = Tensor([1, 2, 3], dtype="float32") + >>> y = Tensor([1, 2, 3], dtype="float32") + >>> F.utils._assert_equal(x, y, maxerr=0) + Tensor([1. 2. 3.], device=xpux:0) - import numpy as np - from megengine import tensor - import megengine.functional as F - - x = tensor([1, 2, 3], np.float32) - y = tensor([1, 2, 3], np.float32) - print(F.utils._assert_equal(x, y, maxerr=0).numpy()) - - Outputs: - - .. testoutput:: - - [1. 2. 3.] """ err = ( abs(expect - actual)