From 4b19409a6f61354f5abf9bc8b260987dfb1d4d79 Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Wed, 22 Apr 2020 09:13:17 +0800 Subject: [PATCH 1/2] add example for maxpool --- mindspore/ops/operations/nn_ops.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 1fb65e3b76..23ddd9f021 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -913,6 +913,11 @@ class MaxPool(_Pool): Outputs: Tensor, with shape :math:`(N, C_{out}, H_{out}, W_{out})`. + + Examples: + >>> input_tensor = Tensor(np.arange(1*3*3*4).reshape((1,3,3,4)),mindspore.float32) + >>> maxpool_op = P.MaxPool(padding="VALID", ksize=2, strides=1) + >>> output_tensor = maxpool_op(input_tensor) """ @prim_attr_register @@ -959,6 +964,11 @@ class MaxPoolWithArgmax(_Pool): - **output** (Tensor) - Maxpooling result, with shape :math:`(N, C_{out}, H_{out}, W_{out})`. - **mask** (Tensor) - Max values' index represented by the mask. + + Examples: + >>> input_tensor = Tensor(np.arange(1*3*3*4).reshape((1,3,3,4)),mindspore.float32) + >>> maxpool_arg_op = P.MaxPoolWithArgmax(padding="VALID", ksize=2, strides=1) + >>> output_tensor, argmax = maxpool_arg_op(input_tensor) """ def __init__(self, ksize=1, strides=1, padding="valid"): super(MaxPoolWithArgmax, self).__init__(ksize, strides, padding) From f3bf699b8db970873b5666f0dcdf4e3096149047 Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Wed, 22 Apr 2020 16:39:56 +0800 Subject: [PATCH 2/2] add example for maxpool --- mindspore/ops/operations/nn_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 23ddd9f021..61a7d6e833 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -915,7 +915,7 @@ class MaxPool(_Pool): Tensor, with shape :math:`(N, C_{out}, H_{out}, W_{out})`. Examples: - >>> input_tensor = Tensor(np.arange(1*3*3*4).reshape((1,3,3,4)),mindspore.float32) + >>> input_tensor = Tensor(np.arange(1 * 3 * 3 * 4).reshape((1, 3, 3, 4)), mindspore.float32) >>> maxpool_op = P.MaxPool(padding="VALID", ksize=2, strides=1) >>> output_tensor = maxpool_op(input_tensor) """ @@ -966,7 +966,7 @@ class MaxPoolWithArgmax(_Pool): - **mask** (Tensor) - Max values' index represented by the mask. Examples: - >>> input_tensor = Tensor(np.arange(1*3*3*4).reshape((1,3,3,4)),mindspore.float32) + >>> input_tensor = Tensor(np.arange(1 * 3 * 3 * 4).reshape((1, 3, 3, 4)), mindspore.float32) >>> maxpool_arg_op = P.MaxPoolWithArgmax(padding="VALID", ksize=2, strides=1) >>> output_tensor, argmax = maxpool_arg_op(input_tensor) """