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

torch.randperm(n, *, out=None, dtype=torch.int64, layout=torch.strided,
               device=None     , requires_grad=False) → LongTensor

返回整数从0到n-1的随机排列。

Parameters:

    n (int)
    上限整数(不包含)

Example:

>>> torch.randperm(4)
tensor([2, 1, 0, 3])

承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。

猜你喜欢

转载自blog.csdn.net/weixin_36670529/article/details/115228872