Pytorch:如何创建一个随机整数张量，其中某个百分比具有某个值？例如，25% 为 1，其余为 0

Question

在 pytorch 中，我可以创建一个随机的零和一个张量，每个张量的分布约为 %50

import torch 
torch.randint(low=0, high=2, size=(2, 5))

我想知道如何制作一个只有 25% 的值为 1，其余为 0 的张量？

Answer 1

您可以使用 rand 在 0,1 之间生成一个随机张量，并将其与 0.25 进行比较:

(torch.rand(size=(2,5)) < 0.25).int()

输出:

tensor([[0, 0, 0, 0, 1],
        [1, 0, 0, 0, 0]], dtype=torch.int32)