From 865987679483622504bac50a96ab279ad714d9f1 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 30 Apr 2020 11:01:13 +0800 Subject: [PATCH] docs(mge/functional): fix doctest GitOrigin-RevId: cd23860ab8c5dbf9853dd00d869e64f13247c0fc --- python_module/megengine/functional/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python_module/megengine/functional/nn.py b/python_module/megengine/functional/nn.py index 93fd66b72..adc616f10 100644 --- a/python_module/megengine/functional/nn.py +++ b/python_module/megengine/functional/nn.py @@ -443,7 +443,7 @@ def one_hot(inp: Tensor, num_classes: int) -> Tensor: import megengine.functional as F inp = tensor(np.arange(1, 4, dtype=np.int32)) - out = F.one_hot(inp) + out = F.one_hot(inp, num_classes=4) print(out.numpy()) Outputs: -- GitLab