xidulu commented on a change in pull request #17302: [numpy]add op
random.logistic, random.gumbel
URL: https://github.com/apache/incubator-mxnet/pull/17302#discussion_r377426714
##########
File path: python/mxnet/symbol/numpy/random.py
##########
@@ -251,6 +252,95 @@ def lognormal(mean=0.0, sigma=1.0, size=None, dtype=None,
ctx=None, out=None):
return _mx_np_symbol.exp(normal(loc=mean, scale=sigma, size=size,
dtype=dtype, ctx=ctx, out=out))
+def logistic(loc=0.0, scale=1.0, size=None, ctx=None, out=None):
+ r"""Draw samples from a logistic distribution.
+
+ Samples are drawn from a logistic distribution with specified
+ parameters, loc (location or mean, also median), and scale (>0).
+
+ Parameters
+ ----------
+ loc : float or array_like of floats, optional
+ Parameter of the distribution. Default is 0.
+ scale : float or array_like of floats, optional
+ Parameter of the distribution. Must be non-negative.
+ Default is 1.
+ size : int or tuple of ints, optional
+ Output shape. If the given shape is, e.g., ``(m, n, k)``, then
+ ``m * n * k`` samples are drawn. If size is ``None`` (default),
+ a single value is returned if ``loc`` and ``scale`` are both scalars.
+ Otherwise, ``np.broadcast(loc, scale).size`` samples are drawn.
+
+ Returns
+ -------
+ out : ndarray or scalar
+ Drawn samples from the parameterized logistic distribution.
+ """
+ from ._symbol import _Symbol as np_symbol
+ input_type = (isinstance(loc, np_symbol), isinstance(scale, np_symbol))
+ if ctx is None:
+ ctx = current_context()
+ if size == ():
+ size = None
+ if input_type == (True, True):
+ return _npi.logistic(loc, scale, loc=None, scale=None, size=size,
+ ctx=ctx, out=out)
+ elif input_type == (False, True):
+ return _npi.logistic(scale, loc=loc, scale=None, size=size,
+ ctx=ctx, out=out)
+ elif input_type == (True, False):
+ return _npi.logistic(loc, loc=None, scale=scale, size=size,
+ ctx=ctx, out=out)
+ else:
+ return _npi.logistic(loc=loc, scale=scale, size=size,
+ ctx=ctx, out=out)
+
+
+def gumbel(loc=0.0, scale=1.0, size=None, ctx=None, out=None):
+ r"""Draw samples from a Gumbel distribution.
+
+ Draw samples from a Gumbel distribution with specified location and
+ scale. For more information on the Gumbel distribution, see
+ Notes and References below.
Review comment:
Remove this sentence.
----------------------------------------------------------------
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
