【PyTorch】rand/randn/randint/randperm的区别
  TEZNKK3IfmPf 2023年11月15日 10 0

问题

随机数的应用场景十分广泛,例如搭建完成网络进行测试的时候需要随机输入,PyTorch提供了rand/randn/randint/randperm等多种随机数的生成方法,那么这些方法的区别是什么呢?在实际开发时,应该如何选择呢?

方法

 import torch 

x = torch.zeros(size=(3, 224, 224))

# [0, n) 随机序列
a = torch.randperm(10)
print(a) # tensor([8, 0, 9, 1, 7, 2, 3, 5, 6, 4])

# [a,b) 均匀分布, 注意size参数接受的是一个tuple,而不是整数
# b = torch.rand(size=1) # rand(): argument 'size' must be tuple of ints, not int
b = torch.rand(size=(1, ))
b0 = torch.rand_like(x)
print(b0.shape)

# [a,b) 标准正太分布
d = torch.randn(size=(1,))

# [a, b) 随机整数
c = torch.randint(1, 10, (1,) )
【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月15日 0

暂无评论

推荐阅读
  TEZNKK3IfmPf   2024年04月19日   18   0   0 python
  I7JaHrFMuDsU   24天前   21   0   0 pytorch
  TEZNKK3IfmPf   2024年04月19日   25   0   0 idepython
TEZNKK3IfmPf