cupy.random.choice#

cupy.random.choice(a, size=None, replace=True, p=None)[源代码]#

返回一个从给定的一维数组中随机取值的数组。

返回数组的每个元素根据 p 或均匀分布独立地从 a 中采样。

注意

目前当 replace=False 时,不支持使用 p 参数。

参数
  • a (1-D array-likeint) – 如果是类数组对象,则从其元素中生成随机样本。如果是一个整数,则生成的随机样本就如同 acupy.arange(n) 一样

  • size (intint 元组) – 数组的形状。

  • replace (布尔值) – 样本是否进行替换抽样。

  • p (1-D array-like) – 与 a 中每个条目关联的概率。如果未给出,则样本假定在 a 中的所有条目上是均匀分布的。

返回

一个 a 值组成的数组,其分布根据 p 或均匀分布。

返回类型

cupy.ndarray

另见

numpy.random.choice()