Paddle
Paddle copied to clipboard
[0-size Tensor Job2 No.17-20、31-34] Add 0-size Tensor support for segment_max
PR Category
Execute Infrastructure
PR Types
Improvements
Description
17 paddle.geometric.segment_max 18 paddle.geometric.segment_mean 19 paddle.geometric.segment_min 20 paddle.geometric.segment_sum
31 paddle.incubate.segment_max 32 paddle.incubate.segment_mean 33 paddle.incubate.segment_min 34 paddle.incubate.segment_sum paddle.incubate.* 算子和paddle.geometric.*相同
修改前向和反向,CPU/GPU 都使用impl infermeta 没有修改
PaddleAPITest 测试numpy error, 其他通过
paddle.geometric.segment_max
GPU
CPU
其他算子相同
paddle.incubate.* 算子和paddle.geometric.*相同,只是增加deprecation 提示
python/paddle/incubate/tensor/math.py