layers.py 文件源码

def conv2d(inputs, filters, kernel_size = 1, strides = 1, pad = 'VALID', name = None):
    """
        Create a Convolutional Layer
        args :
            inputs  : (tensor) input Tensor
            filters     : (int) number of filters
            kernel_size : (int) size of the kernel
            strides     : (int) Value of stride
            pad         : ('VALID'/'SAME')
        return :
            tf.Tensor
    """
    with tf.name_scope(name):
        kernel = tf.Variable(tf.contrib.layers.xavier_initializer(uniform=False)([kernel_size,kernel_size, inputs.get_shape().as_list()[3], filters]), name= 'weights')
        conv = tf.nn.conv2d(inputs, kernel, [1,strides,strides,1], padding=pad, data_format='NHWC')
        with tf.device('/cpu:0'):
            tf.summary.histogram('weights_summary', kernel, collections = ['train'])
        return conv