随机改组火炬张量

Question

我的批量张量为64。批量中的每个样本均为$(System.DefaultWorkingDirectory)形状。因此，我的批次形状为[4, 300]。我想随机调整该批次的元素。换句话说，我想混洗所有64个[64, 4, 300]张量。我怎样才能做到这一点？所得张量显然将具有[4, 300]形状，但是形状为[64, 4, 300]的所有64行将以不同的顺序排列。

Answer 1

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