model.py 文件源码

def _create_conv_layer(fitler_width, in_channels, out_channels):
        kernel_shape = [fitler_width,
                        in_channels,
                        out_channels]
        biases_shape = [out_channels]
        return {
            'weights': tf.get_variable(
                'weights',
                kernel_shape,
                initializer=tf.uniform_unit_scaling_initializer(1.0)),
            'biases': tf.get_variable(
                'biases',
                biases_shape,
                initializer=tf.constant_initializer(0.0))
        }