Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 d7495838 编写于 作者: Z zenghsh3
浏览文件
操作

refine

上级 04a05d1d
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
paddle/fluid/operators/sampling_id_op.h
浏览文件 @ d7495838
...@@ -54,7 +54,7 @@ class SamplingIdKernel : public framework::OpKernel<T> { ...@@ -54,7 +54,7 @@ class SamplingIdKernel : public framework::OpKernel<T> {
static_cast<T>(context.Attr<float>("max"))); static_cast<T>(context.Attr<float>("max")));
std::vector<int64_t> ids(batch_size); std::vector<int64_t> ids(batch_size);
for (size_t i = 0; i < batch_size; ++i) { for (int i = 0; i < batch_size; ++i) {
T r = dist(engine); T r = dist(engine);
int idx = width - 1; int idx = width - 1;
for (int j = 0; j < width; ++j) { for (int j = 0; j < width; ++j) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或