python - 用 Pytorch 随机选择？

Question

我有一个张量的图片，想从中随机选择。我正在寻找相当于 np.random.choice() .

import torch

pictures = torch.randint(0, 256, (1000, 28, 28, 3))

假设我想要 10 张这样的照片。

Answer 1

torch没有 np.random.choice() 的等效实现，见讨论here .另一种方法是使用混洗索引或随机整数进行索引。

用替换来做到这一点:

生成 n 个随机索引

用这些索引索引你的原始张量

pictures[torch.randint(len(pictures), (10,))]  

要做到这一点而无需更换:

洗牌索引

取第 n 个元素

indices = torch.randperm(len(pictures))[:10]

pictures[indices]

阅读更多关于 torch.randint 和 torch.randperm .第二个代码片段的灵感来自这个 post在 PyTorch 论坛中。