mindspore.ops.csr_to_coo

mindspore.ops.csr_to_coo(tensor)

将一个CSRTensor转化成一个COOTensor。

Note

现在只支持2维CSRTensor。

参数：

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

返回：

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

异常：

• TypeError - tensor 不是CSRTensor。

• ValueError - tensor 不是2维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]]