Python Pytorch linspace()用法及代碼示例

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

函數torch.linspace()返回一維步長張量，該步張量在起點和終點之間等距。

輸出張量是尺寸步長的一維。

用法:torch.linspace(start, end, steps=100, out=None)

參數：
start：點集的起始值。
end：點集的最終值
steps：每對相鄰點之間的間隙。默認值：100
out(Tensor, optional)：輸出張量

返回類型：張量

代碼1：

# Importing the PyTorch library 
import torch 
  
# Applying the linspace function and 
# storing the resulting tensor in 't' 
a = torch.linspace(3, 10, 5) 
print("a = ", a) 
  
b = torch.linspace(start =-10, end = 10, steps = 5) 
print("b = ", b)

輸出：

a =  tensor([ 3.0000,  4.7500,  6.5000,  8.2500, 10.0000])
b =  tensor([-10.,  -5.,   0.,   5.,  10.])

代碼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 
  
# Applying the linspace function to get a tensor of size 15 with values from -5 to 5 
a = torch.linspace(-5, 5, 15) 
print(a) 
  
# Plotting 
plt.plot(a.numpy(), np.zeros(a.numpy().shape), color = 'red', marker = "o")  
plt.title("torch.linspace")  
plt.xlabel("X")  
plt.ylabel("Y")  
  
plt.show()

輸出：

tensor([-5.0000, -4.2857, -3.5714, -2.8571, -2.1429, -1.4286, -0.7143,  0.0000,
         0.7143,  1.4286,  2.1429,  2.8571,  3.5714,  4.2857,  5.0000])

[torch.FloatTensor of size 15]

相關用法

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