mindspore_gl.graph.remove_self_loop

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

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

参数：

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

• mode (str, 可选) - 操作矩阵的类型。支持的图类型为 'coo' 和 'dense' 。默认值：'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