Как случайным образом выбрать 5% от общих значений из тензора?

Question

a = torch.rand(2,5,10) Я хочу выбрать не более 5% значений из тензора a случайным образом, а затем умножить эти значения на -1? Как это сделать? любезно, дайте общее c решение, так как форма тензора не фиксирована

Answer 1

Это сработало для меня

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