Browse Source

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

GitOrigin-RevId: 1d74cec7cb
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
1dd27b39d6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/functional/nn.py

+ 1
- 1
imperative/python/megengine/functional/nn.py View File

@@ -668,7 +668,7 @@ def max_pool2d(


Examples: Examples:
>>> import numpy as np >>> 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) >>> F.nn.max_pool2d(input, 2, 1, 0)
Tensor([[[[ 5. 6. 7.] Tensor([[[[ 5. 6. 7.]
[ 9. 10. 11.]]]], device=xpux:0) [ 9. 10. 11.]]]], device=xpux:0)


Loading…
Cancel
Save