pytorch shuffle tensor的函数（打乱张量）

np.random.shuffle能打乱numpy，但是对tensor没用
查阅资料后发现，可以用torch.randperm实现

if __name__ == '__main__':

    # init
    t = torch.tensor([[1,2],[3,4]])
    print(t)

    # shuffle tensor
    idx = torch.randperm(t.nelement())
    t = t.view(-1)[idx].view(t.size())
    print(t)

输出：

tensor([[1, 2],
        [3, 4]])
tensor([[4, 2],
        [1, 3]])