From ada54ba5a02d99ff58a7eb7bee222b3569215512 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 24 Aug 2022 17:23:08 +0800 Subject: [PATCH] docs(mge): update AvgPool2d & functional.nn.avg_pool2d example GitOrigin-RevId: e2859c7f99a52002a483815fc4993712216c0abd --- imperative/python/megengine/functional/nn.py | 5 +++-- imperative/python/megengine/module/pooling.py | 9 +++++++++ 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 4e724838..3899baf5 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -709,8 +709,9 @@ def avg_pool2d( output tensor of shape :math:`(N, C, H_{\text{out}}, W_{\text{out}})`. Examples: - >>> input = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4)) - >>> F.avg_pool2d(input, kernel_size=2, stride=2, padding=[1,0], mode="average") + >>> import numpy as np + >>> inp = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4)) + >>> F.avg_pool2d(inp, kernel_size=2, stride=2, padding=[1,0], mode="average") Tensor([[[[0.25 1.25] [6.5 8.5 ]]]], device=xpux:0) """ diff --git a/imperative/python/megengine/module/pooling.py b/imperative/python/megengine/module/pooling.py index f089ea95..1345e54a 100644 --- a/imperative/python/megengine/module/pooling.py +++ b/imperative/python/megengine/module/pooling.py @@ -88,6 +88,15 @@ class AvgPool2d(_PoolNd): mode: whether to include the padding values while calculating the average, set to "average" will do counting. Default: "average_count_exclude_padding" + + Examples: + >>> import numpy as np + >>> 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) + Tensor([[[[0.25 1.25] + [6.5 8.5 ]]]], device=xpux:0) + """ def __init__(