layers.py 文件源码

def __init__(self, incoming, p=0.5, rescale=True, **kwargs):
        super(InputDropoutLayer, self).__init__(incoming, **kwargs)
        self.p = p
        self.rescale = rescale
        constant_dim = 1
        self.constant_dim = constant_dim
        self.dropoutshape = (self.input_shape[:constant_dim]) + (self.input_shape[constant_dim+1:])
        self.dropoutlayer = lasagne.layers.DropoutLayer(incoming=self.dropoutshape, p=p, rescale=rescale, **kwargs)

        # add parameters to this layer
        self.params.update(self.dropoutlayer.params)

        self.dropoutmask = self.add_param(init.Constant(1), 
                                          self.dropoutshape, 'dropoutmask',
                                          trainable=False, regularizable=False)