這個函數主要是為了生成對角線全1，其余部分全0的二維數組

函數原型：
result = torch.eye(n，m=None，out=None)
參數解釋：
n：行數
m：列數
out：輸出類型
例：
c = torch.eye（3）
print(c)
print(type(c))

輸出

tensor([[1., 0., 0.],
? ? ? ? [0., 1., 0.],
? ? ? ? [0., 0., 1.]])
<class 'torch.Tensor'>

?

總結

以上是生活随笔為你收集整理的pytorch 之 torch.eye()函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。