未验证 提交 6c636102 编写于 作者: T tangwei12 提交者: GitHub

Merge pull request #12892 from seiriosPlus/samplingIdOp

bug fix
...@@ -63,7 +63,7 @@ class SamplingIdKernel : public framework::OpKernel<T> { ...@@ -63,7 +63,7 @@ class SamplingIdKernel : public framework::OpKernel<T> {
break; break;
} }
} }
ids[i] = ins_vector[i * width + idx]; ids[i] = ins_vector[idx];
} }
std::vector<int64_t> out_dim; std::vector<int64_t> out_dim;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册