torch中的spmm

系列文章目录

本系列记录自己的代码学习知识

---

torch.matmul的前后两个矩阵维度不同的小结
torch中的transpose和view的不同

---

前言

---

一、torch中的spmm

我自己运行的结果，spmm就是矩阵乘法

例：

z = y.transpose(0,1)
output = torch.spmm(y, z)
print('z:',z,'y:',y)
print('output:',output)

z: tensor([[1, 1],
        [2, 2],
        [3, 3]]) y: tensor([[1, 2, 3],
        [1, 2, 3]])
output: tensor([[14, 14],
        [14, 14]])

---

总结

结合看到的知识，自己重新整理。

torch.spmm就是乘法