Pytorch：线性代数

线性代数

PyTorch的线性函数主要封装了Blas和Lapack，其用法和接口都与之类似。常用的线性代数函数如表3-7所示。

表3-7: 常用的线性代数函数

函数	功能
trace	对角线元素之和(矩阵的迹)
diag	对角线元素
triu/tril	矩阵的上三角/下三角，可指定偏移量
mm/bmm	矩阵乘法，batch的矩阵乘法
addmm/addbmm/addmv/addr/badbmm..	矩阵运算
t	转置
dot/cross	内积/外积
inverse	求逆矩阵
svd	奇异值分解

具体使用说明请参见官方文档1，需要注意的是，矩阵的转置会导致存储空间不连续，需调用它的.contiguous方法将其转为连续。

---

1. http://pytorch.org/docs/torch.html#blas-and-lapack-operations↩

In [88]:

b = a.t()
b.is_contiguous()

Out[88]:

False

In [89]:

b.contiguous()

Out[89]:

tensor([[ 0.,  9.],
        [ 3., 12.],