haojin2 commented on a change in pull request #15370: [numpy][doc-fix] mean,
transpose, stack, split, log2, rint and radians
URL: https://github.com/apache/incubator-mxnet/pull/15370#discussion_r298426875
##########
File path: python/mxnet/_numpy_op_doc.py
##########
@@ -173,3 +173,107 @@ def _np_cumsum(a, axis=None, dtype=None, out=None):
`axis` is not None or `a` is a 1-d array.
"""
pass
+
+
+def _np_mean(a, axis=None, dtype=None, out=None, keepdims=None):
+ """
+ mean(a, axis=None, dtype=None, out=None, keepdims=None)
+
+ Compute the arithmetic mean along the specified axis.
+ Returns the average of the array elements.
+ The average is taken over the flattened array by default, otherwise over
the specified axis.
+
+ Parameters
+ ----------
+ a : `ndarray`
+ ndarray containing numbers whose mean is desired.
+ axis : None or int or tuple of ints, optional
+ Axis or axes along which the means are computed. The default is to
compute the mean of the flattened array.
+ If this is a tuple of ints, a mean is performed over multiple axes,
+ instead of a single axis or all the axes as before.
+ dtype : data-type, optional
+ Type to use in computing the mean. For integer inputs, the default is
float32;
+ for floating point inputs, it is the same as the input dtype.
+ out : ndarray, optional
Review comment:
Is `out` before `keepdims` or after in our version? I believe our `out`
should always be the second last argument.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to 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
