Python PyTorch clamp()用法及代码示例

PyTorch torch.clamp()方法将所有输入元素限制在[min，max]范围内，并返回结果张量。
Clamp method

用法： torch.clamp(inp, min, max, out=None)

参数

inp:这是输入张量。
min:这是一个数字，指定输入将被钳位到的范围的lower-bound。
max:这是一个数字，指定输入将被钳位到的范围的upper-bound。
out:输出张量。

返回：它返回张量。

让我们借助几个示例来了解这个概念：
范例1：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.randn(6) 
print(a) 
  
# Applying the clamp function and  
# storing the result in 'out' 
out = torch.clamp(a, min = 0.5, max = 0.9) 
print(out)

输出：

 -0.9214
-0.1268
 1.1570
-0.2753
-0.0746
 0.7957
[torch.FloatTensor of size 6]
 0.5000
 0.5000
 0.9000
 0.5000
 0.5000
 0.7957
[torch.FloatTensor of size 6]

范例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([1, 4, 6, 8, 10, 14]) 
print(a) 
  
# Applying the clamp function and  
# storing the result in 'out' 
out = torch.clamp(a, min = 5, max = 10) 
print(out)

输出：

 1
  4
  6
  8
 10
 14
[torch.FloatTensor of size 6]
  5
  5
  6
  8
 10
 10
[torch.FloatTensor of size 6]?

相关用法

注：本文由纯净天空筛选整理自PranchalKatiyar大神的英文原创作品 Python – PyTorch clamp() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。