未验证 提交 002a7b4d 编写于 作者: K kexinzhao 提交者: GitHub

fix scatter op equation (#6304)

上级 16822fb7
...@@ -87,10 +87,15 @@ class ScatterOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -87,10 +87,15 @@ class ScatterOpMaker : public framework::OpProtoAndCheckerMaker {
AddInput("Updates", "The updated value of updates op"); AddInput("Updates", "The updated value of updates op");
AddOutput("Out", "The output of add op"); AddOutput("Out", "The output of add op");
AddComment(R"DOC( AddComment(R"DOC(
Scatter Operator by selecting from the first axis, Scatter Operator.
Out = Ref This operator obtains output by updating the input on selected indices on the first axis:
$$
Out = Ref \\
Out[Index] = Ref[Index] + Updates Out[Index] = Ref[Index] + Updates
$$
)DOC"); )DOC");
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册