如何钳制PyTorch张量的大小？

Question

我知道我可以使用-01-将张量的值限制在最小/最大范围内，但是如果要按幅度（绝对值）进行限制，该怎么办？示例：

torch.clamp

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

Answer 1

这里是一种方法：

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

（注意：这是就地操作）