mindspore.SparseTensor
- class mindspore.SparseTensor(indices, values, shape)[源代码]
用来表示某一张量在给定索引上非零元素的集合。
SparseTensor 只能在 Cell 的构造方法中使用。
说明
此接口从 1.7 版本开始弃用,并计划在将来移除。请使用
mindspore.COOTensor
。对于稠密张量,其 SparseTensor(indices, values, shape) 具有 dense[indices[i]] = values[i] 。
- 参数:
indices (Tensor) - shape为 \((N, ndims)\) 的二维整数张量,其中N和ndims分别表示稀疏张量中 values 的数量和SparseTensor维度的数量。
values (Tensor) - shape为 \((N)\) 的一维张量,其内部可以为任何数据类型,用来给 indices 中的每个元素提供数值。
shape (tuple(int)) - shape为 \((ndims)\) 的整数元组,用来指定稀疏矩阵的稠密形状。
- 返回:
SparseTensor,由 indices 、 values 和 shape 组成。
样例:
>>> import mindspore as ms >>> 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的非零元素值。