torch.randperm

`torch.randperm` 是 PyTorch 中的一个函数，用于生成指定大小的随机排列。其语法为：`torch.randperm(n, *, generator=None, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor`。其中，参数 `n` 表示要生成的随机排列的长度，返回的是一个包含 0 到 n-1 的随机排列的张量。

下面是一个示例：

import torch

# 生成长度为 5 的随机排列
randperm = torch.randperm(5)
print(randperm)

输出：

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

可以看到，返回的张量是一个长度为 5 的随机排列。