提交 b2959589 编写于 作者: M Megvii Engine Team

docs(mge): update AvgPool2d example

GitOrigin-RevId: 2d5e51505587386a4318371747a8a470aa885fda
上级 ada54ba5
...@@ -94,6 +94,7 @@ class AvgPool2d(_PoolNd): ...@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册