PyTorch还支持一些线性函数,免得用起来的时候自己造轮子,具体用法参考官方文档。如下表所示:
函数 | 功能 |
trace | 对⻆角线元素之和(矩阵的迹) |
diag | 对⻆角线元素 |
triu/tril | 矩阵的上三⻆角/下三⻆角,可指定偏移量量 |
mm/bmm | 矩阵乘法, batch的矩阵乘法 |
addmm/addbmm/addmv/addr/badbmm.. | 矩阵运算 |
t | 转置 |
dot/cross | 内积/外积 |
inverse | 求逆矩阵 |
svd | 奇异值分解 |
PyTorch中的 Tensor 支持超过一百种操作,包括转置、索引、切片、数学运算、线性代数、随机数等等,可参考官方文档
未经允许不得转载!PyTorch的一些线性代数函数