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]]