| from __future__ import annotations | |
| from cupy.random import _generator | |
| def shuffle(a): | |
| """Shuffles an array. | |
| Args: | |
| a (cupy.ndarray): The array to be shuffled. | |
| .. seealso:: :meth:`numpy.random.shuffle` | |
| """ | |
| rs = _generator.get_random_state() | |
| return rs.shuffle(a) | |
| def permutation(a): | |
| """Returns a permuted range or a permutation of an array. | |
| Args: | |
| a (int or cupy.ndarray): The range or the array to be shuffled. | |
| Returns: | |
| cupy.ndarray: If `a` is an integer, it is permutation range between 0 | |
| and `a` - 1. | |
| Otherwise, it is a permutation of `a`. | |
| .. seealso:: :meth:`numpy.random.permutation` | |
| """ | |
| rs = _generator.get_random_state() | |
| return rs.permutation(a) | |