haojin2 commented on a change in pull request #17316: [NumPy] add op
random.laplace
URL: https://github.com/apache/incubator-mxnet/pull/17316#discussion_r366766965
##########
File path: python/mxnet/symbol/numpy/random.py
##########
@@ -290,6 +290,54 @@ def choice(a, size=None, replace=True, p=None, ctx=None,
out=None):
return _npi.choice(p, a=a, size=size, replace=replace, ctx=ctx,
weighted=True, out=out)
+def laplace(loc=0.0, scale=1.0, size=None, dtype=None, ctx=None, out=None):
+ r"""Draw random samples from a Laplace distribution.
+
+ Samples are distributed according to a Laplace distribution parametrized
+ by *loc* (mean) and *scale* (the exponential decay).
+
+ Parameters
+ ----------
+ loc : float, The position of the distribution peak.
+
+ scale : float, the exponential decay.
+
+ 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. Default is None, in
which case a single value is returned.
+
+ dtype : {'float16', 'float32', 'float64'}, optional
+ Data type of output samples. Default is 'float32'
+ ctx : Context, optional
+ Device context of output. Default is current context.
+ out : ``ndarray``, optional
+ Store output to an existing ``ndarray``.
+
+ Returns
+ -------
+ out : _Symbol (symbol representing `mxnet.numpy.ndarray` in computational
graphs)
+ Drawn samples from the parameterized Laplace distribution.
+ """
+ from ._symbol import _Symbol as np_symbol
+ input_type = (isinstance(loc, np_symbol), isinstance(scale, np_symbol))
+ if dtype is None:
+ dtype = 'float32'
+ if ctx is None:
+ ctx = current_context()
+ if size == ():
+ size = None
+ if input_type == (True, True):
+ return _npi.laplace(loc, scale, loc=None, scale=None, size=size,
+ ctx=ctx, dtype=dtype, out=out)
Review comment:
alignment
----------------------------------------------------------------
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:
us...@infra.apache.org
With regards,
Apache Git Services
