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

PyTorch是由Facebook開發的開源機器學習庫。它用於深度神經網絡和自然語言處理。

函數torch.acos()為PyTorch中的反餘弦函數提供支持。它期望輸入在[-1，1]範圍內，並以弧度形式給出輸出。如果輸入不在[-1，1]範圍內，則返回nan。輸入類型為張量，如果輸入包含多個元素，則將計算按元素的反餘弦值。

用法: torch.acos(x, out=None)

參數：
x：輸入張量
name(可選)：輸出張量

返回類型：與x具有相同類型的張量。

代碼1：

# Importing the PyTorch library 
import torch 
  
# A constant tensor of size 6 
a = torch.FloatTensor([1.0, -0.5, 3.4, 0.2, 0.0, -2]) 
print(a) 
  
# Applying the inverse cos function and 
# storing the result in 'b' 
b = torch.acos(a) 
print(b)

輸出：

tensor([ 1.0000, -0.5000,  3.4000,  0.2000,  0.0000, -2.0000])
tensor([0.0000, 2.0944,    nan, 1.3694, 1.5708,    nan])

代碼2：可視化

# Importing the PyTorch library 
import torch 
  
# Importing the NumPy library 
import numpy as np 
  
# Importing the matplotlib.pylot function 
import matplotlib.pyplot as plt 
  
# A vector of size 15 with values from -1 to 1 
a = np.linspace(-1, 1, 15) 
  
# Applying the inverse cosine function and 
# storing the result in 'b' 
b = torch.acos(torch.FloatTensor(a)) 
  
print(b) 
  
# Plotting 
plt.plot(a, b.numpy(), color = 'red', marker = "o")  
plt.title("torch.acos")  
plt.xlabel("X")  
plt.ylabel("Y")  
  
plt.show()

輸出：

tensor([3.1416, 2.6005, 2.3664, 2.1790, 2.0137, 1.8605, 1.7141, 1.5708, 1.4274,
        1.2810, 1.1279, 0.9626, 0.7752, 0.5411, 0.0000])

相關用法

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