423 Star 4.3K Fork 424

GVPPaddlePaddle / Paddle

 / 详情

topk运算结果中,indices结果不正确

已完成
创建于  
2021-10-09 14:05

源自github用户mamingjie-China:
当topk的输入中,出现相同大小的值时,运算结果中,indices的结果在GPU和CPU版本的Paddle中不一致,且不同机器,均在CPU版本下,运算结果也不相同。初步看了一下代码,应该是在排序的时候,所使用的排序算法为不稳定排序所致。
示例代码
import paddle.fluid as fluid
import numpy as np
def gen_data():
return {
"input": np.array([[[2, 2, 2, 2], [3,3,2,3], [4, 4, 3, 4]], [[5, 5, 6, 6], [7,7,7,7], [6, 7, 10, 8]]], dtype="float32")
}

input = fluid.data(name="input", shape=[2,3,4], dtype='float32')
z = fluid.layers.topk(input=input, k=2, name='z')
place = fluid.CPUPlace()
exe = fluid.Executor(place)
print(gen_data()["input"].shape)
print(gen_data()["input"])
z_value, z_index = exe.run(feed=gen_data(),
fetch_list=[z[0].name, z[1].name])
print(z_value,"###############", z_index)
CPU结果
image
GPU结果
image

评论 (1)

PaddlePaddle-Gardener 创建了任务
展开全部操作日志

源自github用户Aurelius84:
感谢反馈。经确认,topk的CPUKernel底层实现调用了std::partial_sort,这个算法是不稳定的。我们会修复这个问题。

PaddlePaddle-Coordinator 任务状态待办的 修改为已完成

登录 后才可以发表评论

状态
负责人
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
开始日期   -   截止日期
-
置顶选项
优先级
参与者(1)
Python
1
https://gitee.com/paddlepaddle/Paddle.git
git@gitee.com:paddlepaddle/Paddle.git
paddlepaddle
Paddle
Paddle

搜索帮助