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

docs(mge): update functional.nn.max_pool2d docstring

GitOrigin-RevId: 1d74cec7cba1e3ae2c129e962c441f47b877b38d
上级 6e54a3bf
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册