Python PyTorch clamp用法及代碼示例

本文簡要介紹python語言中 torch.clamp 的用法。

用法:

torch.clamp(input, min=None, max=None, *, out=None) → Tensor

參數：

• input(Tensor) -輸入張量。

• min(數字或者Tensor,可選的) -lower-bound 要鉗製到的範圍

• max(數字或者Tensor,可選的) -upper-bound 要鉗製到的範圍

關鍵字參數：

out(Tensor,可選的) -輸出張量。

將 input 中的所有元素夾在 [ min 、 max ] 範圍內。讓 min_value 和 max_value 分別為 min 和 max ，這將返回：

y_i = \min(\max(x_i, \text{min\_value}_i), \text{max\_value}_i)

如果 min 是 None ，則沒有下限。或者，如果 max 是 None，則沒有上限。

注意

如果 min 大於 max torch.clamp(..., min, max) 將 input 中的所有元素設置為 max 的值。

例子：

>>> a = torch.randn(4)
>>> a
tensor([-1.7120,  0.1734, -0.0478, -0.0922])
>>> torch.clamp(a, min=-0.5, max=0.5)
tensor([-0.5000,  0.1734, -0.0478, -0.0922])

>>> min = torch.linspace(-1, 1, steps=4)
>>> torch.clamp(a, min=min)
tensor([-1.0000,  0.1734,  0.3333,  1.0000])