haojin2 commented on a change in pull request #15956: [Numpy] random.randint() implemented URL: https://github.com/apache/incubator-mxnet/pull/15956#discussion_r316483418
########## File path: python/mxnet/ndarray/numpy/random.py ########## @@ -20,7 +20,67 @@ from ...context import current_context from . import _internal as _npi -__all__ = ['uniform'] + +__all__ = ['randint', 'uniform'] + + +def randint(low, high=None, size=None, dtype=None, **kwargs): + """Return random integers from `low` (inclusive) to `high` (exclusive). + + Return random integers from the "discrete uniform" distribution of + the specified dtype in the "half-open" interval [`low`, `high`). If + `high` is None (the default), then results are from [0, `low`). + + Parameters + ---------- + low : int + Lowest (signed) integer to be drawn from the distribution (unless + ``high=None``, in which case this parameter is one above the + *highest* such integer). + high : int, optional + If provided, one above the largest (signed) integer to be drawn + from the distribution (see above for behavior if ``high=None``). + 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 : dtype, optional + Desired dtype of the result. All dtypes are determined by their + name, i.e., 'int64', 'int', etc, so byteorder is not available + and a specific precision may have different C types depending + on the platform. The default value is 'np.int'. Review comment: Remember to add `ctx` and `out` to the `Parameters` section, same for all other docs for this op ---------------------------------------------------------------- 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
