個人python與dl學習常見常用函式

NorthernDeerherd發表於2024-08-15

torch庫
tensor(張量)函式:
torch.empty(5,3) 直接建立一個53的tensor,但未初始化
torch.rand(5.3) 直接建立一個5
3的tensor,但是其中的數值隨機初始化
torch.arrange(x)/(x,y)/(x,y,z) 建立一維張量,0x-1/xy-1/x~y-1,間隔為z
torch.reshape() 將張量維度進行修改,如(3,4),則修改為3行4列二維張量
x.shape 獲取維度

相關文章