作用
将输入input张量的每个元素的夹紧到区间 [min,max],并返回结果得到一个新张量
参数及定义
import torch
torch.clamp(input, min, max, out = None)
参数:
input (Tensor) – 输入张量
min (Number) – 限制范围下限
max (Number) – 限制范围上限
out (Tensor, optional) – 输出张量
运算规则
|-min, if x_i < min
|
y_i = | x_i, if min <= x_i <= max
|
|-max, if x_i > max
示例
import torch
a = torch.randint(low = 0, high = 100, size = (1, 10))
print(a)
a = torch.clamp(a, 10, 20)
print(a)
>>tensor([[ 8, 92, 17, 60, 83, 81, 33, 53, 12, 79]])
>>tensor([[10, 20, 17, 20, 20, 20, 20, 20, 12, 20]])