Browse Source

add AvgPooling layer

tags/v0.2.0-alpha
zhaojichen 6 years ago
parent
commit
beca38a399
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/nn/layer/pooling.py

+ 2
- 2
mindspore/nn/layer/pooling.py View File

@@ -258,7 +258,7 @@ class AvgPool1d(_PoolNd):
[[[[8. 8. 7. 4.]
[8. 4. 0. 9.]
[6. 4. 6. 1.]
[6. 8. 8. 5.]
[6. 8. 8. 5.]]
[[4. 8. 5. 4.]
[8. 4. 7. 5.]
[3. 5. 3. 9.]
@@ -270,7 +270,7 @@ class AvgPool1d(_PoolNd):
[[[[7.6640625 6.3320312]
[4. 4.3320312]
[5.3320312 3.6660156]
[7.3320312 7 ]
[7.3320312 7 ]]
[[5.6640625 5.6640625]
[6.3320312 5.3320312]
[3.6660156 5.6640625]


Loading…
Cancel
Save