search

PaddlePaddle / PaddleCustomDevice

PaddlePaddle custom device implementaion. (『飞桨』自定义硬件接入实现)
Apache License 2.0
70 stars 148 forks source link

[MLU] Fix reduce_mean bug #1349

Closed Birdylx closed 3 months ago

Birdylx commented 3 months ago

fix reduce mean bug in this case：

x = paddle.randn([2, 3])
mea = x.mean(keepdim=True) # should be with shape [1, 1], but got error
paddle-bot[bot] commented 3 months ago

Thanks for your contribution!