Преобразование scipy coo_matrix в разреженный тензор pytorch

Question

У меня есть coo_matrix:

from scipy.sparse import coo_matrix
coo = coo_matrix((3, 4), dtype = "int8")

, который я хочу преобразовать в разреженный тензор pytorch.Согласно документации https://pytorch.org/docs/master/sparse.html он должен соответствовать формату coo, но я не могу найти простой способ сделать преобразование.Любая помощь будет принята с благодарностью!

Answer 1

Используя данные, как в Документах Pytorch , это можно сделать, просто используя атрибуты Numpy coo_matrix:

import torch
import numpy as np
from scipy.sparse import coo_matrix

coo = coo_matrix(([3,4,5], ([0,1,1], [2,0,2])), shape=(2,3))

values = coo.data
indices = np.vstack((coo.row, coo.col))

i = torch.LongTensor(indices)
v = torch.FloatTensor(values)
shape = coo.shape

torch.sparse.FloatTensor(i, v, torch.Size(shape)).to_dense()

Вывод

0 0 3
4 0 5
[torch.FloatTensor of size 2x3]