Add comments

release/0.13.0
fengjiayi 8 years ago
parent d44dbc4a52
commit 20c8ff0f5f

@ -32,12 +32,18 @@ class RandomCropOp : public framework::OperatorWithKernel {
class RandomCropOpMaker : public framework::OpProtoAndCheckerMaker { class RandomCropOpMaker : public framework::OpProtoAndCheckerMaker {
public: public:
void Make() override { void Make() override {
AddInput("X", ""); AddInput("X", "A batch of instances to random crop.");
AddOutput("Out", ""); AddInput("Seed", "The random seed.");
AddInput("Seed", ""); AddOutput("Out", "The cropped instance batch.");
AddOutput("SeedOut", "").AsDispensable(); AddOutput("SeedOut", "The random seed after random cropping.")
AddAttr<std::vector<int>>("shape", ""); .AsDispensable();
AddComment(""); AddAttr<std::vector<int>>("shape", "The shape of a cropped instance.");
AddComment(R"DOC(
This operator takes a batch of instance, and do random cropping on each instance.
It means that cropping positions differs on each instance, which is determined
by an uniform random generator. All cropped instances have the same shape, which
is determined by the operator's attribute 'shape'.
)DOC");
} }
}; };

Loading…
Cancel
Save