mindspore_gl.graph.remove_self_loop

mindspore_gl.graph.remove_self_loop(adj, mode='dense')

从输入矩阵对象中删除对角矩阵，可以选择对dense矩阵或coo格式的矩阵进行操作。

参数：

• adj (scipy.sparse.coo) - 目标矩阵。

• mode (str) - 操作矩阵的类型。默认值：dense。

返回：

移除对角矩阵后的对象。 如果 mode 为dense，返回Tensor类型；如果 mode 为coo返回spy.sparse.coo类型。

支持平台：

Ascend GPU

样例：

>>> from mindspore_gl.graph.self_loop import remove_self_loop
>>> import scipy.sparse as sp
>>> adj = sp.csr_matrix(([1, 2, 3, 4], ([0, 1, 2, 2], [0, 1, 2, 1])), shape=(3, 3)).tocoo()
>>> adj = remove_self_loop(adj, 'coo')
>>> print(adj)
(1, 2)        4