小编zah*_*awi的帖子

是否可以在 PyTorch 中按行对两个 2D 张量进行打乱，但保持两者的顺序相同？我知道您可以使用以下代码按行对 2D 张量进行洗牌：

a=a[torch.randperm(a.size()[0])]

详细说明：如果我有 2 个张量

a = torch.tensor([[1, 1, 1, 1, 1],
            [2, 2, 2, 2, 2],
            [3, 3, 3, 3, 3]])

b = torch.tensor([[4, 4, 4, 4, 4],
            [5, 5, 5, 5, 5],
            [6, 6, 6, 6, 6]])

并通过一些函数/代码块运行它们以随机洗牌但保持相关性并产生如下所示的内容

a = torch.tensor([[2, 2, 2, 2, 2],
            [1, 1, 1, 1, 1],
            [3, 3, 3, 3, 3]])

b = torch.tensor([[5, 5, 5, 5, 5],
            [4, 4, 4, 4, 4],
            [6, 6, 6, 6, 6]]) …