mindspore.ops.dense_to_sparse_coo

mindspore.ops.dense_to_sparse_coo(tensor)[源代码]

将常规Tensor转为稀疏化的COOTensor。

Note

现在只支持2维Tensor。

参数：

  • tensor (Tensor) - 一个稠密Tensor，必须是2维。

返回：

返回一个2维的COOTensor，是原稠密Tensor的稀疏化表示。分为：

  • indices (Tensor) - 二维整数张量，其中N和ndims分别表示稀疏张量中 values 的数量和COOTensor维度的数量。

  • values (Tensor) - 一维张量，用来给 indices 中的每个元素提供数值。

  • shape (tuple(int)) - 整数元组，用来指定稀疏矩阵的稠密形状。目前只支持2维Tensor输入，所以 shape 长度只能为2。

异常：

  • TypeError - tensor 不是Tensor。

  • ValueError - tensor 不是2维Tensor。

支持平台：

GPU

样例：

>>> from mindspore import Tensor
>>> import mindspore as ms
>>> x = Tensor([[1, 0], [-5, 0]], ms.float32)
>>> output = ops.dense_to_sparse_coo(x)
>>> print(output)