mindspore.Tensor.to_csr

mindspore.Tensor.to_csr()

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

说明

现在只支持二维Tensor。

返回：

返回一个二维的CSRTensor，是原稠密Tensor的稀疏化表示。其中数据分别为：

• indptr (Tensor) - 一维整数张量，表示每行非零元素的在 values 中存储的起止位置。

• indices (Tensor) - 一维整数张量，表示每个元素的列索引值。

• values (Tensor) - 一维张量，用来表示索引对应的数值。

• shape (tuple(int)) - 整数元组，用来指定稀疏矩阵的稠密形状。

异常：

• ValueError - Tensor的shape不是二维。

支持平台：

GPU

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,  0], [-5, 0]]), mindspore.float32)
>>> output = x.to_csr()
>>> print(output.indptr, output.indices, output.values, output.shape)
[0 1 2] [0 0] [ 1. -5.] (2, 2)