Python PyTorch div()用法及代碼示例

PyTorch torch.div()方法將輸入的每個元素除以一個常量，然後返回一個新的修改過的張量。
Clamp method

用法： torch.div(inp, other, out=None)

參數

inp:這是輸入張量。
other:這是一個要劃分為輸入inp的每個元素的數字。
out:輸出張量。

返回：它返回張量。

讓我們借助幾個示例來了解這個概念：
範例1：

# 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 div function and  
# storing the result in 'out' 
out = torch.div(a, 0.5) 
print(out)

輸出：

1
  4
  6
  8
 10
 14
[torch.FloatTensor of size 6]
  2
  8
 12
 16
 20
 28
[torch.FloatTensor of size 6]

範例2：

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

輸出：

 -0.8453
-0.1101
 0.9431
-0.3041
 1.4305
-0.0390
[torch.FloatTensor of size 6]
-2.8176
-0.3669
 3.1436
-1.0137
 4.7683
-0.1300
[torch.FloatTensor of size 6]

相關用法

注：本文由純淨天空篩選整理自PranchalKatiyar大神的英文原創作品 Python – PyTorch div() method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。