coalesce

paddle.sparse. coalesce ( x, name=None ) [源代码]

coalesce 操作包含排序和合并相同 indices 两步,执行 coalesce 后,x 变成按 indices 进行有序排序,并行每个 indices 只出现一次。

参数

  • x (Tensor) - 输入 SparseCooTensor

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

返回 coalesce 后的 SparseCooTensor。

代码示例

import paddle

indices = [[0, 0, 1], [1, 1, 2]]
values = [1.0, 2.0, 3.0]
sp_x = paddle.sparse.sparse_coo_tensor(indices, values)
sp_x = paddle.sparse.coalesce(sp_x)
print(sp_x.indices())
#[[0, 1], [1, 2]]
print(sp_x.values())
#[3.0, 3.0]