pytorch - 如何从张量中随机选择 5% 的总值？

Question

a = torch.rand(2,5,10) 我想从张量中a随机选择最多 5% 的值，然后将这些值乘以-1? 怎么做？请给出一个通用解决方案，因为张量的形状不固定

标签： pytorchtensor

Answer 1

这对我有用

out = (torch.rand_like(a) - 0.05).sign().type_as(a) * a