core.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:kfs 作者: the-moliver 项目源码 文件源码
def __init__(self, filters,
                 sum_axes,
                 filter_axes,
                 activation=None,
                 use_bias=True,
                 kernel_initializer='glorot_uniform',
                 bias_initializer='zeros',
                 kernel_activation=None,
                 kernel_regularizer=None,
                 bias_regularizer=None,
                 activity_regularizer=None,
                 kernel_constraint=None,
                 bias_constraint=None,
                 **kwargs):
        if 'input_shape' not in kwargs and 'input_dim' in kwargs:
            kwargs['input_shape'] = (kwargs.pop('input_dim'),)
        super(FilterDims, self).__init__(**kwargs)
        self.kernel_initializer = initializers.get(kernel_initializer)
        self.bias_initializer = initializers.get(bias_initializer)
        self.activation = activations.get(activation)
        self.kernel_activation = activations.get(kernel_activation)
        self.filters = filters
        self.sum_axes = list(sum_axes)
        self.sum_axes.sort()
        self.filter_axes = list(filter_axes)
        self.filter_axes.sort()
        self.kernel_regularizer = regularizers.get(kernel_regularizer)
        self.bias_regularizer = regularizers.get(bias_regularizer)
        self.activity_regularizer = regularizers.get(activity_regularizer)
        self.kernel_constraint = constraints.get(kernel_constraint)
        self.bias_constraint = constraints.get(bias_constraint)
        self.use_bias = use_bias
        self.input_spec = InputSpec(min_ndim=2)
        self.supports_masking = True
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号