tf.layers.conv2d和tf.nn.conv2d使用区别

来源:互联网 发布:安能物流鲁班系统mac 编辑:程序博客网 时间:2024/06/08 17:39

看了很多tensorflow卷积的例子,有的用了tf.nn.conv2d,有的用了tf.layers.conv2d,那么这两个究竟有啥不同呢?


tf.layers.conv2d(inputs, filters, kernel_size, strides=(1,1),

                            padding='valid', data_format='channels_last',
                    dilation_rate=(1,1), activation=None,
                    use_bias=True, kernel_initializer=None,
                    bias_initializer=init_ops.zeros_initializer(),
                    kernel_regularizer=None,
                    bias_regularizer=None,
                    activity_regularizer=None, trainable=True,
                    name=None, reuse=None)


tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu=None, data_format=None, name=None)


对于卷积来说,作用是一样的。tf.layers.conv2d 使用tf.nn.convolution作为后端。

一个参数要注意一下:

filters: 是一个4维张量,其type必须和输入一样,

    [filter_height, filter_width, in_channels, out_channels]filter:一个整数,输出空间的维度,也就是卷积核的数量。
原创粉丝点击