From 1dd27b39d64e550c11a736a83a345d10f22461e9 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 2 Sep 2022 11:53:17 +0800 Subject: [PATCH] docs(mge): update functional.nn.max_pool2d docstring GitOrigin-RevId: 1d74cec7cba1e3ae2c129e962c441f47b877b38d --- imperative/python/megengine/functional/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index f1b466e0..d2acbc1b 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -668,7 +668,7 @@ def max_pool2d( Examples: >>> import numpy as np - >>> input = tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4)) + >>> input = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4)) >>> F.nn.max_pool2d(input, 2, 1, 0) Tensor([[[[ 5. 6. 7.] [ 9. 10. 11.]]]], device=xpux:0)