CoalesceSparse
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR/Ascend 950DT |
√ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 |
√ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
√ |
| Atlas 200I/500 A2 推理产品 |
× |
| Atlas 推理系列产品 |
× |
| Atlas 训练系列产品 |
× |
功能说明
- 算子功能：将相同坐标点的value进行累加求和，进而减少Coo_Tensor的内存大小。
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| unique_len |
输入 |
去重后的索引数。 |
INT32、INT64 |
ND |
| unique_indices |
输入 |
去重后的索引数组。 |
INT32、INT64 |
ND |
| indices |
输入 |
索引数组。 |
INT32、INT64 |
ND |
| values |
输入 |
每个坐标对应的元素值。 |
INT32、FLOAT16、FLOAT32 |
ND |
| new_indices |
输出 |
合并后的索引数组。 |
INT32、INT64 |
ND |
| new_values |
输出 |
合并后的元素值。 |
INT32、FLOAT16、FLOAT32 |
ND |
约束说明
重索引后的indices值不能超过int32上限。
调用说明
