Pytorch 按某个维度打乱数据方法

目录

1、采用的主要函数

2、一维数据打乱

3、二维数据打乱

4、多维数据打乱

---

主要是用来解决pytorch数据的打乱问题。或则针对pytorch的某个数组维度进行打乱。

1、采用的主要函数

        该函数，随机返回0~n-1个整数。返回值连续，且混乱。

torch.randperm(n, out=None, dtype=torch.int64, layout=torch.strided, device=None, requires_grad=False)

2、一维数据打乱

x = torch.arange(0, 10)
 
print(x)
b=torch.randperm(x.size(0))
print(x[b])

3、二维数据打乱

a=torch.rand(3,5)
print(a)
 
a=a[torch.randperm(a.size(0))]
print(a)
 
a=a[:,torch.randperm(a.size(1))]
print(a)

4、多维数据打乱

a=torch.rand(3,5,5)
print(a)
 
a=a[torch.randperm(a.size(0))]
print(a)
 
a=a[:,torch.randperm(a.size(1)),:]
print(a)