maskConv.py 文件源码

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

项目:yoctol-keras-layer-zoo 作者: Yoctol 项目源码 文件源码
def compute_mask(self, inputs, mask=None):
        dimension = K.ndim(inputs)
        mask_tensor = K.any(K.not_equal(inputs, self.mask_value), axis=-1)
        mask_base = K.any(mask_tensor, axis=1, keepdims=True)
        for axis in range(2, dimension - 1):
            mask_axis = K.any(mask_tensor, axis=axis, keepdims=True)
            mask_base = tf.logical_and(mask_base, mask_axis)
        return mask_base
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号