PyTorch的一些线性代数函数

PyTorch还支持一些线性函数,免得用起来的时候自己造轮子,具体用法参考官方文档。如下表所示:

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


PyTorch中的 Tensor 支持超过一百种操作,包括转置、索引、切片、数学运算、线性代数、随机数等等,可参考官方文档


 

未经允许不得转载!PyTorch的一些线性代数函数