fix scatter op equation (#6304)

release/0.11.0
kexinzhao 8 years ago committed by GitHub
parent 16822fb702
commit 002a7b4d01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -87,10 +87,15 @@ class ScatterOpMaker : public framework::OpProtoAndCheckerMaker {
AddInput("Updates", "The updated value of updates op");
AddOutput("Out", "The output of add op");
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
$$
)DOC");
}
};

Loading…
Cancel
Save