cupy.random.choice

cupy.random.choice(a, size=None, replace=True, p=None)[source]

Returns an array of random values from a given 1-D array.

Each element of the returned array is independently sampled from a according to p or uniformly.

Note

Currently p is not supported when replace=False.

Parameters:
  • a (1-D array-like or int) – If an array-like, a random sample is generated from its elements. If an int, the random sample is generated as if a was cupy.arange(n)
  • size (int or tuple of ints) – The shape of the array.
  • replace (boolean) – Whether the sample is with or without replacement.
  • p (1-D array-like) – The probabilities associated with each entry in a. If not given the sample assumes a uniform distribution over all entries in a.
Returns:

An array of a values distributed according to

p or uniformly.

Return type:

cupy.ndarray

See also

numpy.random.choice()