Python torch 模块,randperm() 实例源码

2022-09-02 13:31:39 浏览数 (1)

代码语言: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])

0 人点赞