megengine.random.shuffle¶
- shuffle(inp)¶
打乱张量的内容原地修改张量。这个函数只是沿着第一个轴打乱张量。张量顺序改变但不改变值
- 参数
inp (
Tensor
) – 输入一个张量
实际案例
import numpy as np import megengine as mge import megengine.random as rand x = mge.tensor(np.arange(10)) rand.shuffle(x) print(x.numpy()) y = mge.tensor(np.arange(18)).reshape(6,3) rand.shuffle(y) print(y.numpy())
输出:
[7 9 3 0 8 2 4 5 6 1] [[12. 13. 14.] [ 3. 4. 5.] [15. 16. 17.] [ 0. 1. 2.] [ 9. 10. 11.] [ 6. 7. 8.]]