pytorch 常用線性函式詳解
阿新 • • 發佈：2020-01-16
Pytorch的線性函式主要封裝了Blas和Lapack，其用法和介面都與之類似。
常用的線性函式如下：
| 函式 | 功能 |
| trace | 對角線元素之和（矩陣的跡） |
| diag | 對角線元素 |
| triu/tril | 矩陣的上三角/下三角，可指定偏移量 |
| mm/bmm | 矩陣乘法，batch的矩陣乘法 |
| t | 轉置 |
| dot/cross | 內積/外積 |
| inverse | 求逆矩陣 |
| svd | 奇異值分解 |
注意：矩陣的轉置會使儲存空間不連續，需呼叫它的.contiguous方法轉為連續。
例如：
import torch as t b=a.t() b.is_contiguous() 輸出：False b=b.contiguous() b.is_contiguous() 輸出：True
以上這篇pytorch 常用線性函式詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。