diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 2c1f9888282188b8066924cef0108ee697f91da6..48c6bb99bbc82710a5fc42f2bbc076acc7fe656e 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -1183,14 +1183,17 @@ def conv2d(input,
act=None,
name=None):
"""
- **Convlution2D Layer**
-
The convolution2D layer calculates the output based on the input, filter
- and strides, paddings, dilations, groups parameters. Input(Input) and
- Output(Output) are in NCHW format. Where N is batch size, C is the number of
+ and strides, paddings, dilations, groups parameters. Input and
+ Output are in NCHW format, where N is batch size, C is the number of
channels, H is the height of the feature, and W is the width of the feature.
- The details of convolution layer, please refer UFLDL's `convolution,
- `_ .
+ Filter is in MCHW format, where M is the number of output image channels,
+ C is the number of input image channels, H is the height of the filter,
+ and W is the width of the filter. If the groups is greater than 1,
+ C will equal the number of input image channels divided by the groups.
+ Please refer to UFLDL's `convolution
+ `_
+ for more detials.
If bias attribution and activation type are provided, bias is added to the
output of the convolution, and the corresponding activation function is
applied to the final result.
@@ -1201,15 +1204,14 @@ def conv2d(input,
Out = \sigma (W \\ast X + b)
- In the above equation:
+ Where:
* :math:`X`: Input value, a tensor with NCHW format.
* :math:`W`: Filter value, a tensor with MCHW format.
* :math:`\\ast`: Convolution operation.
* :math:`b`: Bias value, a 2-D tensor with shape [M, 1].
* :math:`\\sigma`: Activation function.
- * :math:`Out`: Output value, the shape of :math:`Out` and :math:`X` may be
- different.
+ * :math:`Out`: Output value, the shape of :math:`Out` and :math:`X` may be different.
Example:
@@ -1220,6 +1222,7 @@ def conv2d(input,
Filter shape: :math:`(C_{out}, C_{in}, H_f, W_f)`
- Output:
+
Output shape: :math:`(N, C_{out}, H_{out}, W_{out})`
Where
@@ -1231,7 +1234,7 @@ def conv2d(input,
Args:
input (Variable): The input image with [N, C, H, W] format.
- num_filters(int): The number of filter. It is as same as the output
+ num_filters(int): The number of filter. It is as same as the output
image channel.
filter_size (int|tuple|None): The filter size. If filter_size is a tuple,
it must contain two integers, (filter_size_H, filter_size_W).
@@ -1254,7 +1257,8 @@ def conv2d(input,
bias_attr (ParamAttr): Bias parameter for the Conv2d layer. Default: None
use_cudnn (bool): Use cudnn kernel or not, it is valid only when the cudnn
library is installed. Default: True
- use_mkldnn (bool): Use mkldnn kernels or not.
+ use_mkldnn (bool): Use mkldnn kernels or not, it is valid only when compiled
+ with mkldnn library. Default: False
act (str): Activation type. Default: None
name (str|None): A name for this layer(optional). If set None, the layer
will be named automatically.