diff --git a/tensorflow2fluid/doc/tf.layers.conv2d.md b/tensorflow2fluid/doc/tf.layers.conv2d.md index e46edb9d65284d0a5f89f97dc7459a3842acb37b..6d7b422c1caa049b0c009c306f77610baab61186 100644 --- a/tensorflow2fluid/doc/tf.layers.conv2d.md +++ b/tensorflow2fluid/doc/tf.layers.conv2d.md @@ -51,7 +51,7 @@ TensorFlow: 默认输入数据格式为`NHWC`,表示`(batch,height, width, i | 输入 | 卷积核 | 输出 | |--------------------|-------------------|------------------| |NHWC | (kernel_h, kernel_w, filters_num, in_channels)| (batch, out_h, out_w, filters_num)| -|NDHW | (kernel_h, kernel_w, filters_num, in_channels) | (batch, filters_num, out_h, out_w)| +|NCHW | (kernel_h, kernel_w, filters_num, in_channels) | (batch, filters_num, out_h, out_w)| PaddlePaddle:只支持输入数据格式为`NCHW`,且**卷积核格式**与TensorFlow不同,其中输入、输出、卷积核对应关系如下表所示, @@ -84,4 +84,4 @@ PaddlePaddle: 使用`paddle.fluid.layers.conv2d`,可参考 # 卷积核Shape: (5, 3, 4, 4) inputs = paddle.fluid.layers.data(dtype='float32', shape=[3, 200, 200], name='inputs) pad_inputs = paddle.fluid.layers.pad2d(inputs, paddings=[1, 2, 1, 2]) -outputs = paddle.fluid.layers.conv2d(pad_inputs, 5, [4, 4], (1, 1)) \ No newline at end of file +outputs = paddle.fluid.layers.conv2d(pad_inputs, 5, [4, 4], (1, 1))