megengine.random.shuffle¶

shuffle(inp)¶

打乱张量的内容原地修改张量。这个函数只是沿着第一个轴打乱张量。张量顺序改变但不改变值

参数: inp (Tensor) – 输入一个张量
返回: None.

实际案例

>>> import numpy as np
>>> import megengine.random as rand
>>> x = mge.tensor(np.arange(10))
>>> rand.shuffle(x)
>>> x.numpy()   
array([4, 5, 9, 6, 2, 8, 1, 0, 3, 7], dtype=int32)
>>> y = mge.tensor(np.arange(18)).reshape(6,3)
>>> rand.shuffle(y)
>>> y.numpy()   
array([[ 3,  4,  5],
       [ 6,  7,  8],
       [15, 16, 17],
       [ 0,  1,  2],
       [12, 13, 14],
       [ 9, 10, 11]], dtype=int32)

megengine.random.permutation

megengine.random.normal