mars.tensor.random.randint#

mars.tensor.random.randint(low, high=None, size=None, dtype='l', density=None, chunk_size=None, gpu=None)[来源]#

返回从 low（包含）到 high（不包含）的随机整数。

返回指定dtype在“半开”区间 [low, high) 的 “离散均匀” 分布的随机整数。如果high 为 None（默认值），则结果来自 [0, low)。

Parameters

  • low (int) – 从分布中抽取的最低（有符号）整数（除非 high=None，在这种情况下，此参数是该 最高整数之上的一个）。

  • high (int, 可选) – 如果提供，表示从分布中绘制的最大（带符号）整数之上的一个。

  • size (int 或 tuple 的 ints, 可选) – 输出形状。 如果给定的形状是，例如，(m, n, k)，那么 m * n * k 个样本将被抽取。 默认值是 None，在这种情况下返回一个 单一值。

  • dtype (数据类型, 可选) – 结果所需的 dtype。所有 dtype 都由其名称确定，即 ‘int64’, ‘int’ 等，因此 byteorder 不可用，并且特定精度可能根据平台具有不同的 C 类型。默认值是 ‘np.int’。

  • 密度 (浮点数, 可选) – 如果指定了密度，将创建一个稀疏张量

  • chunk_size (int 或 tuple 的 int 或 tuple 的 ints, 可选) – 每个维度上所需的块大小

  • gpu (bool, 可选) – 如果为True，则在GPU上分配张量，默认为False

  • dtype – 返回张量的数据类型。

Returns

out – size形状的随机整数张量，来自适当的分布，或者如果没有提供size，则为单个随机整数。

Return type

int 或整数的张量

另请参阅

random.random_integers

类似于 randint，仅适用于闭合区间 [low, high]，如果省略 high，则1是最低值。特别地，这个函数是用来生成均匀分布的离散非整数。

示例

>>> import mars.tensor as mt

>>> mt.random.randint(2, size=10).execute()
array([1, 0, 0, 0, 1, 1, 0, 0, 1, 0])
>>> mt.random.randint(1, size=10).execute()
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])

生成一个范围在0到4（包含0和4）之间的2 x 4整型张量：

>>> mt.random.randint(5, size=(2, 4)).execute()
array([[4, 0, 2, 1],
       [3, 2, 2, 0]])