代码语言:javascript复制
torch.randperm(n, *, out=None, dtype=torch.int64, layout=torch.strided,
device=None , requires_grad=False) → LongTensor
返回整数从0到n-1的随机排列。
Parameters:
n (int) 上限整数(不包含)
Example:
代码语言:javascript复制>>> torch.randperm(4)
tensor([2, 1, 0, 3])