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

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

函數torch.eye()返回a返回大小為n * m的2-D張量，對角線為1，其他位置為零。

用法:torch.eye(n, m, out=None)

代碼1：

# Importing the PyTorch library 
import torch 
  
  
# Applying the eye function and 
# storing the resulting tensor in 'a' 
a = torch.eye(3, 4) 
print("a = ", a) 
  
b = torch.eye(3, 3) 
print("b = ", b) 
  
c = torch.eye(5, 1) 
print("c = ", c)

輸出：

a =  tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])
b =  tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])
c =  tensor([[1.],
        [0.],
        [0.],
        [0.],
        [0.]])