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