提交 b20586cf 编写于 作者: L liuwei1031 提交者: phlrain

scatter_op bug fix, test=develop (#16866)

* scatter_op bug fix, test=develop

* scatter_op bug fix, test=develop

* scatter_op bug fix, test=develop
上级 9b30c51e
...@@ -42,10 +42,6 @@ class ScatterOp : public framework::OperatorWithKernel { ...@@ -42,10 +42,6 @@ class ScatterOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(ctx->GetInputDim("Updates")[0], PADDLE_ENFORCE_EQ(ctx->GetInputDim("Updates")[0],
ctx->GetInputDim("Ids")[0], ctx->GetInputDim("Ids")[0],
"Updates and Ids should have same batch-size."); "Updates and Ids should have same batch-size.");
framework::DDim data_dim(updates_dims);
for (int i = 1; i < data_dim.size(); ++i) {
PADDLE_ENFORCE_EQ(data_dim[i], updates_dims[i]);
}
ctx->SetOutputDim("Out", ref_dims); ctx->SetOutputDim("Out", ref_dims);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册