Created by: AshburnLee
PR types
New features
PR changes
OPs
Describe
为unique OP 添加了CUDA实现,函数较多,给出函数间调用关系。 3个二值比较函数作为其他函数的参数,未在下表示出:
Compute(){
未指名轴
UniqueFlattendCUDATensor(){
ComputeUniqueFlatten()
}
指明轴
UniqueDimsCUDATensor(){
1. Transpose & reshape
2. Calculate 'sorted_indices', 'inverse', 'counts'
ComputeUniqueDims()
3. Select indices and reshape back to get 'out'
IndexSelect()
}
}