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