随机洗牌火炬张量

Question

我有一个批量大小为 64 的张量。批量中的每个样本的形状为[4, 300]。所以，我的批次形状是[64, 4, 300]。我想随机洗牌批次的元素。换句话说，我想对所有 64 个[4, 300]张量进行洗牌。我怎样才能做到这一点？显然，得到的张量的形状为[64, 4, 300]，但所有 64 行形状为[4, 300]，将按不同的顺序排列。

Answer 1

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