Pytorch：基于比率/百分比/比率创建张量掩码

Question

如何生成特定比率为 0 和 1 的掩码张量？例如，5 x 10 张量中 0 和 1 的 70:30 将生成：

[[0,0,0,0,0,0,0,0,1,0],
[1,1,1,0,0,0,1,1,0,0],
[0,0,1,0,0,1,1,0,1,0],
[0,1,0,1,0,1,1,0,0,0],
[0,0,0,1,0,0,0,0,0,0]]

Answer 1

您可以使用随机抽样来近似该比率：

s1
s2
s3

平均而言，import torch mask = torch.rand(5, 10) # uniformly distributed between 0 and 1 mask = mask < 0.3 # 30% pixels "on" 将具有适量的“开启”像素。

---

或者，如果您必须恰好有 30% 的“开启”像素，您可以使用 torch.randperm 来随机置换具有精确“开启”像素数量的蒙版：

mask