Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add AvgPooling layer

This commit is contained in:
zhaojichen 2020-04-21 23:50:45 -04:00
parent beca38a399
commit 06ca226da4
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/nn/layer/pooling.py
View File

@ -255,14 +255,6 @@ class AvgPool1d(_PoolNd):
Examples: Examples:
>>> pool = nn.AvgPool1d(kernel_size=3, strides=1) >>> pool = nn.AvgPool1d(kernel_size=3, strides=1)
>>> x = Tensor(np.random.randint(0, 10, [1, 2, 4, 4]), mindspore.float32) >>> x = Tensor(np.random.randint(0, 10, [1, 2, 4, 4]), mindspore.float32)
[[[[8. 8. 7. 4.]
[8. 4. 0. 9.]
[6. 4. 6. 1.]
[6. 8. 8. 5.]]
[[4. 8. 5. 4.]
[8. 4. 7. 5.]
[3. 5. 3. 9.]
[7. 5. 4. 7.]]]]
>>> output = pool(x) >>> output = pool(x)
>>> output.shape() >>> output.shape()
(1, 2, 4, 2) (1, 2, 4, 2)