pytorch_sparse教程
Coalesce
torch_sparse.coalesce(index, value, m, n, op="add") -> (torch.LongTensor, torch.Tensor)
逐行排序index
并删除重复项。通过将重复项映射到一起来删除重复项。对于映射,可以使用任何一种torch_scatter
操作。
参数
- index(LongTensor) -稀疏矩阵的索引张量。
- value (Tensor) -稀疏矩阵的值张量。
- m (int) -稀疏矩阵的第一个维度。
- n (int) -稀疏矩阵的第2维。
- op (string,可选)-使用的散射操作。(默认值:“add”)
返回
- index (LongTensor) -稀疏矩阵的合并索引张量。
- value (Tensor) -稀疏矩阵的合并值张量。
import torch
from torch_sparse import coalesce
index = torch.tensor([[1, 0, 1, 0, 2, 1],
[0, 1, 1, 1, 0, 0]])
value = torch.Tensor([[1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7]])
index, value = coalesce(index, value, m=3, n=2)
------------------------------------------------------------
Transpose
torch_sparse.transpose(index, value, m, n) -> (torch.LongTensor, torch.Tensor)
对稀疏矩阵的0维和1维进行转置。
参数
- index(LongTensor) -稀疏矩阵的索引张量。
- value (Tensor) -稀疏矩阵的值张量。
- m (int) -稀疏矩阵的第一个维度。
- n (int) -稀疏矩阵的第2维。coalesced (bool,可选)-如果设置为False,将不会合并输出。(默认值是真实的)
返回
- index(longtensor) - 稀疏基质的转置索引张量。
- value(张量) - 稀疏矩阵的转置值张量。
import torch
from torch_sparse import transpose
index = torch.tensor([[1, 0, 1, 0, 2, 1], [0, 1, 1, 1, 0, 0]])
value = torch.Tensor([[1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7]])
index, value = transpose(index, value, 3, 2)
----------------------------------------------------
print(index)
tensor([[0, 0, 1, 1],
[1, 2, 0, 1]])
print(value)
tensor([[7.0, 9.0],
[5.0, 6.0],
[6.0, 8.0],
[3.0, 4.0]])
Sparse Dense Matrix Multiplication
torch_sparse.spmm(index, value, m, n, matrix) -> torch.Tensor
一个稀疏矩阵与一个密集矩阵的矩阵乘积。
参数
- index(LongTensor) -稀疏矩阵的索引张量。
- value (Tensor) -稀疏矩阵的值张量。
- m (int) -稀疏矩阵的第一个维度。
- n (int) -稀疏矩阵的第2维。矩阵(张量)-密集矩阵。
返回
- out(张量)-密集输出矩阵。
import torch
from torch_sparse import spmm
index = torch.tensor([[0, 0, 1, 2, 2], [0, 2, 1, 0, 1]])
value = torch.Tensor([1, 2, 4, 1, 3])
matrix = torch.Tensor([[1, 4], [2, 5], [3, 6]])
out = spmm(index, value, 3, 3, matrix)
------------------------------------------
print(out)
tensor([[7.0, 16.0],
[8.0, 20.0],
[7.0, 19.0]])
Sparse Sparse Matrix Multiplication
torch_sparse.spspmm(indexA, valueA, indexB, valueB, m, k, n) -> (torch.LongTensor, torch.Tensor)
两个稀疏张量的矩阵乘积。两个输入稀疏矩阵都需要合并(使用coalesced属性强制)。
参数
- indexA (LongTensor) -第一个稀疏矩阵的索引张量。
- valueA (Tensor) -第一个稀疏矩阵的值张量。
- indexB (LongTensor) -第二稀疏矩阵的索引张量。
- valueB (Tensor) -第二稀疏矩阵的值张量。
- m (int) -第一个稀疏矩阵的第一个维度。
- k (int) -第一个稀疏矩阵的第二次维和第二个稀疏矩阵的第一次维。
- n (int) -第2个稀疏矩阵的第2维。
- coalesced (bool,可选):如果设置为True,将合并两个输入稀疏矩阵。(默认值:False)
返回
- index(LongTensor) -稀疏矩阵的输出索引张量。
- value (Tensor) -稀疏矩阵的输出值张量。
import torch
from torch_sparse import spspmm
indexA = torch.tensor([[0, 0, 1, 2, 2], [1, 2, 0, 0, 1]])
valueA = torch.Tensor([1, 2, 3, 4, 5])
indexB = torch.tensor([[0, 2], [1, 0]])
valueB = torch.Tensor([2, 4])
indexC, valueC = spspmm(indexA, valueA, indexB, valueB, 3, 3, 2)
参考
https://github.com/rusty1s/pytorch_sparse