如何限制 PyTorch 張量的大小？

Question

我知道我可以使用torch.clamp將張量的值限制在某個最小值/最大值內，但是如果我想按幅度（絕對值）限制，我該怎么做？ 例子：

import torch
t = torch.tensor([-5.0, -250, -1, 0.003, 7, 1238])
min_mag = 1 / 10
max_mag = 100

# desired output:
tensor([  -5.0000, -100.0000,   -1.0000,    0.1000,    7.0000,  100.0000])

Answer 1

這是一種方法：

sign = t.sign()
t = t.abs_().clamp_(min_mag, max_mag)
t *= sign

（注意：這是使用就地操作）

Answer 2

一種方法是將元素的符號乘以絕對元素的鉗位版本，如下所示：

output = torch.sign(t) * torch.clamp(torch.abs(t), min_mag, max_mag)