From b2959589fcccecc7e8e7ff6e776bc75b3bb5ba1d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 30 Aug 2022 11:40:52 +0800 Subject: [PATCH] docs(mge): update AvgPool2d example GitOrigin-RevId: 2d5e51505587386a4318371747a8a470aa885fda --- imperative/python/megengine/module/pooling.py | 1 + 1 file changed, 1 insertion(+) diff --git a/imperative/python/megengine/module/pooling.py b/imperative/python/megengine/module/pooling.py index 1345e54a..3d86ab66 100644 --- a/imperative/python/megengine/module/pooling.py +++ b/imperative/python/megengine/module/pooling.py @@ -94,6 +94,7 @@ class AvgPool2d(_PoolNd): >>> m = M.AvgPool2d(kernel_size=2, stride=2, padding=[1,0], mode="average") >>> inp = mge.tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4)) >>> output = m(inp) + >>> output Tensor([[[[0.25 1.25] [6.5 8.5 ]]]], device=xpux:0)