mindspore.ops.SparseToDense

查看源文件
class mindspore.ops.SparseToDense[源代码]

将稀疏Tensor转换为密集Tensor。

输入:
  • indices (Tensor) - 二维Tensor,表示元素在稀疏Tensor中的位置。支持int32、int64,每个元素值都应该是非负的。shape是 \((n, 2)\)

  • values (Tensor) - 一维Tensor,表示 indices 位置上对应的值。shape为 \((n,)\)

  • sparse_shape (tuple(int)) - 指定稀疏Tensor的shape,由两个正整数组成,表示稀疏Tensor的shape为 \((N, C)\)

输出:

Tensor,计算后的Tensor。数据类型与 values 相同,shape由 sparse_shape 所指定。

异常:
  • TypeError - 如果 indices 的数据类型既不是int32也不是int64。

  • ValueError - 如果 sparse_shapeindicesvalues 的shape不符合参数中所描述支持的数据类型。

支持平台:

CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> indices = Tensor([[0, 1], [1, 2]])
>>> values = Tensor([1, 2], dtype=mindspore.float32)
>>> sparse_shape = (3, 4)
>>> sparse_to_dense = ops.SparseToDense()
>>> out = sparse_to_dense(indices, values, sparse_shape)
>>> print(out)
[[0. 1. 0. 0.]
 [0. 0. 2. 0.]
 [0. 0. 0. 0.]]