mindspore.ops.csr_to_coo

mindspore.ops.csr_to_coo(tensor: CSRTensor)

将一个CSRTensor转化成一个COOTensor。

说明

现在只支持二维CSRTensor。

参数：

• tensor (CSRTensor) - 一个CSR矩阵，必须是二维。

返回：

返回一个二维的COOTensor，是原COOTensor的CSR格式表示。

异常：

• TypeError - tensor 不是CSRTensor。

• ValueError - tensor 不是二维CSRTensor。

支持平台：

Ascend GPU CPU

样例：

>>> from mindspore import Tensor, CSRTensor
>>> indptr = Tensor([0, 1, 2]).astype("int32")
>>> indices = Tensor([0, 1]).astype("int32")
>>> values = Tensor([2, 1]).astype("float32")
>>> shape = (2, 4)
>>> x = CSRTensor(indptr, indices, values, shape)
>>> output = ops.csr_to_coo(x)
>>> print(output.indices)
[[0 0]
[1 1]]