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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。