pytorch邻接矩阵

在 PyTorch 中，邻接矩阵通常用稀疏张量（Sparse Tensor）来表示，可以使用 torch.sparse 模块来创建和操作稀疏张量。

首先，我们需要定义一个邻接矩阵，可以使用 scipy.sparse 库中的 csr_matrix 函数来创建。假设我们要创建一个 3x3 的邻接矩阵，其中第 i 行第 j 列的元素为 1 表示节点 i 和节点 j 之间有一条边：

import torch
import scipy.sparse as sp

# 创建一个 3x3 的邻接矩阵
adj = sp.csr_matrix(([1, 1, 1], ([0, 1, 2], [1, 2, 0])), shape=(3, 3))
print(adj.toarray())
# 输出：
# [[0 1 0]
#  [0 0 1]
#  [1 0 0]]

然后，我们可以将邻接矩阵转换为稀疏张量：

# 将邻接矩阵转换为稀疏张量
adj = torch.tensor(adj.todense())
adj = adj.to_sparse()
print(adj)
# 输出：
# tensor(indices=tensor([[0, 1, 2],
#                        [1, 2, 0]]),
#        values=tensor([1, 1, 1]),
#        size=(3, 3), nnz=3, layout=torch.sparse_coo)

在上面的代码中，我们首先使用 `todense()` 函数将邻接矩阵转换为稠密矩阵，然后使用 PyTorch 的 `tensor()` 函数将其转换为张量，最后使用 `to_sparse()` 函数将其转换为稀疏张量。稀疏张量的 `indices` 属性包含了非零元素的坐标，`values` 属性包含了这些非零元素的值。