mindspore.SparseTensor

class mindspore.SparseTensor(indices, values, shape)[源代码]

用来表示某一张量在给定索引上非零元素的集合。

SparseTensor 只能在 Cell 的构造方法中使用。

说明

此接口从 1.7 版本开始弃用，并计划在将来移除。请使用 COOTensor。

对于稠密张量，其 SparseTensor(indices, values, shape) 具有 dense[indices[i]] = values[i] 。

参数：

indices (Tensor) - 形状为 \((N, ndims)\) 的二维整数张量，其中N和ndims分别表示稀疏张量中 values 的数量和SparseTensor维度的数量。
values (Tensor) - 形状为 \((N)\) 的一维张量，其内部可以为任何数据类型，用来给 indices 中的每个元素提供数值。
shape (tuple(int)) - 形状为ndims的整数元组，用来指定稀疏矩阵的稠密形状。

返回：

SparseTensor，由 indices 、 values 和 shape 组成。

样例：

>>> import mindspore as ms
>>> import mindspore.nn as nn
>>> from mindspore import Tensor, SparseTensor
>>> indices = Tensor([[0, 1], [1, 2]])
>>> values = Tensor([1, 2], dtype=ms.float32)
>>> shape = (3, 4)
>>> x = SparseTensor(indices, values, shape)
>>> print(x.values)
[1. 2.]
>>> print(x.indices)
[[0 1]
 [1 2]]
>>> print(x.shape)
(3, 4)

property indices: 返回SparseTensor的索引值。

property shape: 返回SparseTensor的稠密shape。

property values: 返回SparseTensor的非零元素值。