haojin2 edited a comment on issue #11829: Insufficient documentation for GlobalAvgPool1D URL: https://github.com/apache/incubator-mxnet/issues/11829#issuecomment-406446091 @sravanbabuiitm Thanks for reporting the insufficient doc, I agree that we should provide a more detailed doc at https://mxnet.incubator.apache.org/api/python/gluon/nn.html#mxnet.gluon.nn.GlobalAvgPool1D. I also noticed several things to help you understand the behavior of the code: 1. You're using GlobalAvgPool1D(), so if you divide the result you get from np.sum by 5 you get 0.008933034, which should appear on the first row of the result you get on your example. 2. The default value for argument `layout` is saying that the default layout is `NCW`, so in your (1,3,5) sample input the axis 1 is recognized as the channel axis and axis 2 is the data axis, so the average operation is performed on the last axis. That's why you're getting a (1,3,1) tensor as the result. Please let me know if you have further questions or concerns. We'll work on addressing the problem with the doc at the same time.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
