coalesce¶
coalesce 操作包含排序和合并相同 indices 两步,执行 coalesce 后,x 变成按 indices 进行有序排序,并行每个 indices 只出现一次。
返回¶
返回 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]