随机洗牌 Torch 张量

4
我有一个批次大小为64的张量。批次中的每个样本形状为[4, 300]。 因此,我的批次形状为[64, 4, 300]。 我想随机洗牌批次中的元素。换句话说,我想打乱所有64个[4, 300]张量。我该怎么做?生成的张量显然将具有[64, 4, 300]的形状,但是所有[4, 300]的64行将以不同的顺序排序。
1个回答

6
indexes = torch.randperm(data.shape[0])
data = data[indexes]

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接