pytorch稀疏矩阵乘法

PyTorch中的稀疏矩阵乘法可以使用SparseTensor实现。SparseTensor既可以与密集（dense）矩阵做乘法，也可以与稀疏矩阵做乘法。具体实现方法如下：

# Sparse-Dense Matrix Multiplication
x = torch.rand(7, 4)
out = adj.matmul(x)
print(out.shape) # torch.Size([7, 4])

# Sparse-Sparse Matrix Multiplication
adj = adj.matmul(adj)

除了上述方式，SparseTensor还能从其它形式来进行创建，例如从密集矩阵（常见的普通矩阵）进行创建、创建指定大小的单位阵、从scipy矩阵进行创建等。更多创建方式可以参考官方文档。同时，对于几种不同格式的稀疏矩阵，Sparse稀疏矩阵主要存储格式总结一文有更详细的介绍，可以进一步了解。