溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇文章給大家分享的是有關Pytorch 中怎么實現多維數組運算，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。
1、python 的基本索引規則
從 0 開始
對于給定的范圍，如 b = a[m:n], 那么 b 為由 （n-m）個數據組成的新數組，由 a[m],a[m+1],...,a[n-1] 構成。(若 n<m, 得到空)
2、單個 tensor 運算，使用 dim 參數
torch 中對 tensor 的操作方法，若不加 dim 參數表示對整體的 tensr 進行操作，若增加 dim 參則表示按維操作。
例：
a = [[1,2],[3,4],[5,6]] (torch.tensor) torch.mean(a) => 3.5 torch.mean(a,dim=0) => [1.5, 3.5, 5.5] torch.mean(a,dim=1) => [[3],[4]] torch.mean(a,dim=0) => [3,4] torch.mean(a,dim=1) => [1.5, 3.5, 5.5]
注: torch.mean() 是一個降維的操作，所以不會出現在取均值后保持跟原 Tensor 同維的情況。 dim 參數存在時降一維，不存在時得到的是整個 Tensor 的均值。
3、兩個 tensor 運算，構造對應形狀
以乘法為例：
3.1 矩陣乘向量
a = [[1,2],[3,4],[5,6]] b = [1,1]
計算乘法 c = a@b
若 a 拓展為 （N,3,2） N 為 batch_size, 計算 c2 = a@b
若 a,b 同時拓展， 變成（N, 2）,那么需要做一個變換 b = b.view(N,2,1),計算 c3 = a@b
3.2 矩陣乘矩陣
a = [[1,2],[3,4],[5,6]] b =[ [1,1],[1,1]]
計算乘法 c = a@b
若 a 拓展為 （N,3,2） N 為 batch_size, 計算 c2 = a@b
若 a,b 同時拓展， 變成（N, 2, 2）,計算 c3 = a@b
以上就是Pytorch 中怎么實現多維數組運算，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
