活動
Pytorch Tensors
學習重點
PyTorch 是一個開源的機器學習框架,由 Facebook 的研究團隊開發。
特點:
動態計算圖、
強大的自動微分功能、
支持深度學習和強化學習。
PyTorch 由於其靈活性和控制性而經常受到研究人員的青睞。
建立張量:torch.empty()、torch.zeros()、torch.ones()、torch.rand()
張量形狀:
torch.*_like() 方法產生相同維度的張量,
使用 .shape 屬性,取得張量的維度。
資料型別:可以透過 dtype 參數設定張量的資料型別,使用 .to() 是另一種改變資料型別的方法。
PyTorch 張量運算:
torch.zeros(2, 2) + 1
torch.ones(2, 2) * 2
(torch.ones(2, 2) * 7 - 1) / 2