查看: 1460|回复: 0

Pytorch:随机抽样一个向量

[复制链接]

27

主题

37

帖子

116

积分

论坛管理

Rank: 4

积分
116
发表于 2018-10-15 11:17:29 | 显示全部楼层 |阅读模式
  1. sample_size = 5
  2. a = torch.randn(10)
  3. b = torch.randperm(10)
  4. a = a.index_select(0,b)
  5. a = a[0:sample_size]
复制代码
在Matlab中的等价就像是
  1. sample_size = 5
  2. a = rand(10,1)
  3. idx = randperm(10)
  4. b = a(idx(1:sample_size))
复制代码



回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表