如何用其他pytorch函数替换torch.sparse?

How to replace torch.sparse with other pytorch function?

我想用其他 Pytorch 函数替换 torch.sparse 函数。

i = torch.LongTensor([[0, 1, 1], [2, 0, 2]])
v = torch.FloatTensor([3, 4, 5])
out1 = torch.sparse.FloatTensor(i, v, torch.Size([2,3])).to_dense()
out2 = ???
out1 == out2
> tensor(True)

背景:

我正在将 Pytorch 模型转换为 CoreML,但 torch.norm 函数中定义的 sparse_coo_tensor 运算符未使用 CoreMLTools 实现。

少数人遇到了这个问题,但 CoreMLTools 仍然不受支持。所以我想在没有 torch.sparse.FloatTensor.

中使用的运算符的情况下替换它

我试过torch.sparse_coo_tensor但不支持。

我创建了一个简单的 colaboratory notebook 来复制它。请使用以下 colab 对其进行测试。 https://colab.research.google.com/drive/1TzpeJqEcmCy4IuXhhl6LChFocfZVaR1Q?usp=sharing

我之前在Whosebug上询问过不同的运算符,所以请参考。

如果我对sparse_coo格式理解正确的话,i的两行就是复制v值的坐标。这意味着您可以改为创建矩阵,例如:

def dense_from_coo(i, v):
    rows = i[0].max()+1
    cols = i[1].max()+1
    out = torch.zeros(rows, cols)
    out[i[0], i[1]] = v
    return out

print(dense_from_coo(i,v))
>>> tensor([[0., 0., 3.],
            [4., 0., 5.]])

print(torch.sparse.FloatTensor(i, v, torch.Size([2,3])).to_dense())
>>> tensor([[0., 0., 3.],
            [4., 0., 5.]])