Browse Source

docs(mge): update AvgPool2d example

GitOrigin-RevId: 2d5e515055
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
b2959589fc
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/megengine/module/pooling.py

+ 1
- 0
imperative/python/megengine/module/pooling.py View File

@@ -94,6 +94,7 @@ class AvgPool2d(_PoolNd):
>>> m = M.AvgPool2d(kernel_size=2, stride=2, padding=[1,0], mode="average") >>> 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)) >>> inp = mge.tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
>>> output = m(inp) >>> output = m(inp)
>>> output
Tensor([[[[0.25 1.25] Tensor([[[[0.25 1.25]
[6.5 8.5 ]]]], device=xpux:0) [6.5 8.5 ]]]], device=xpux:0)




Loading…
Cancel
Save