当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.random.randint用法及代码示例


用法:

mxnet.ndarray.random.randint(low, high, shape=_Null, dtype=_Null, ctx=None, out=None, **kwargs)

参数

  • low(int, required) - 输出间隔的下边界。生成的所有值都将大于或等于低。
  • high(int, required) - 输出区间的上边界。生成的所有值都将小于高。
  • shape(int or tuple of ints, optional) - 要绘制的样本数。如果形状是,例如,(m, n)lowhigh是标量,输出形状将是(m, n).
  • dtype({'int32', 'int64'}, optional) - 输出样本的数据类型。默认为‘int32’
  • ctx(mxnet.context.Context, optional) - 输出的设备上下文。默认为当前上下文。被覆盖low.contextlow是一个 NDArray。
  • out(NDArray, optional) - 将输出存储到现有的 NDArray。

返回

dtype 类型的 NDArray。如果输入 shape 具有形状,例如 (m, n) ,则返回的 NDArray 的形状将为 (m, n) 。返回的 NDArray 的内容将是来自区间 [low, high) 的样本。

返回类型

ND阵列

从离散的均匀分布中抽取随机样本。

样本在半开区间 [low, high) 上均匀分布(包括 low ,但不包括 high )。

例子

>>> mx.nd.random.randint(5, 100)
[ 90]
<NDArray 1 @cpu(0)
>>> mx.nd.random.randint(-10, 2, ctx=mx.gpu(0))
[ -8]
<NDArray 1 @gpu(0)>
>>> mx.nd.random.randint(-10, 10, shape=(2,))
[ -5  4]
<NDArray 2 @cpu(0)>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.random.randint。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。