mindspore.ops.dense_to_sparse_csr
- mindspore.ops.dense_to_sparse_csr(tensor: Tensor)[源代码]
将常规Tensor转为稀疏化的CSRTensor。
说明
现在只支持二维Tensor。
- 参数:
tensor (Tensor) - 一个稠密Tensor,必须是二维。
- 返回:
返回一个二维的CSRTensor,是原稠密Tensor的稀疏化表示。其中数据分别为:
indptr (Tensor) - 一维整数张量,表示每行非零元素的在 values 中存储的起止位置。
indices (Tensor) - 一维整数张量,表示每个元素的列索引值。
values (Tensor) - 一维张量,用来表示索引对应的数值。
shape (tuple(int)) - 整数元组,用来指定稀疏矩阵的稠密形状。
- 异常:
TypeError - tensor 不是Tensor。
ValueError - tensor 不是二维Tensor。
- 支持平台:
GPU
样例:
>>> from mindspore import Tensor, ops >>> import mindspore as ms >>> x = Tensor([[1, 0], [-5, 0]], ms.float32) >>> output = ops.dense_to_sparse_csr(x) >>> print(output.indptr) [0 1 2] >>> print(output.indices) [0 0] >>> print(output.shape) (2, 2)