Fix the doc.

update-doc-pybind
hedaoyuan 8 years ago
parent b4ba35caeb
commit 656f775c29

@ -69,15 +69,17 @@ class Conv2DOpMaker : public framework::OpProtoAndCheckerMaker {
"Filter", "Filter",
"The filter tensor of convolution operator." "The filter tensor of convolution operator."
"The format of the filter tensor is MCHW, where M is the number of " "The format of the filter tensor is MCHW, where M is the number of "
"output " "output image channels, C is the number of input image channels, "
"image channels, C is the number of input image channels, H and W is " "H and W is height and width of filter. "
"height and width of filter."); "If the groups attribute is greater than 1, C equal the number of "
"input image channels divided by the groups.");
AddOutput("Output", AddOutput("Output",
"The output tensor of convolution operator." "The output tensor of convolution operator."
"The format of output tensor is also NCHW."); "The format of output tensor is also NCHW.");
AddComment(R"DOC( AddComment(R"DOC(
The convolution operation calculates the output based on The convolution operation calculates the output based on the input, filter
the input, filter and strides, paddings parameters. and strides, paddings, groups parameters. The size of each dimension of the
parameters is checked in the infer-shape.
)DOC"); )DOC");
AddAttr<std::vector<int>>("strides", "strides of convolution operator."); AddAttr<std::vector<int>>("strides", "strides of convolution operator.");
AddAttr<std::vector<int>>("paddings", "paddings of convolution operator."); AddAttr<std::vector<int>>("paddings", "paddings of convolution operator.");

Loading…
Cancel
Save