docs(mge/functional): fix doctest

GitOrigin-RevId: cd23860ab8c5dbf9853dd00d869e64f13247c0fc

docs(mge/functional): fix doctest
GitOrigin-RevId: cd23860ab8c5dbf9853dd00d869e64f13247c0fc
86598767 · Megvii Engine Team · Xinran Xu · 4500faf1 · 86598767
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python_module/megengine/functional/nn.py python_module/megengine/functional/nn.py +1 -1

未找到文件。
--- 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: