From 92695a0da8a908a9ee86fb44827ef6b359484783 Mon Sep 17 00:00:00 2001 From: zhaojichen Date: Tue, 21 Apr 2020 04:44:33 -0400 Subject: [PATCH] add AvgPooling layer --- mindspore/nn/layer/pooling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/pooling.py b/mindspore/nn/layer/pooling.py index 299891232e..fef9494ea4 100644 --- a/mindspore/nn/layer/pooling.py +++ b/mindspore/nn/layer/pooling.py @@ -218,7 +218,7 @@ class AvgPool1d(_PoolNd): Typically the input is of shape :math:`(N_{in}, C_{in}, H_{in}, W_{in})`, AvgPool1d outputs regional average in the :math:`(W_{in})`-dimension. Given kernel size - :math:`ks = (w_{ker})` and stride :math:`s = (s_0)`, the operation is as follows. + :math:`ks = w_{ker}` and stride :math:`s = s_0`, the operation is as follows. .. math:: \text{output}(N_i, C_j, h_k, w) = \frac{1}{w_{ker}} \sum_{n=0}^{w_{ker}-1}