Skip to content

Instantly share code, notes, and snippets.

@udibr
Created May 28, 2016 22:52
Show Gist options
  • Save udibr/676c742c8843fdcfdfd24f4dcdc3bdfb to your computer and use it in GitHub Desktop.
Save udibr/676c742c8843fdcfdfd24f4dcdc3bdfb to your computer and use it in GitHub Desktop.
Keras layer to remove mask information
from keras.layers.core import Lambda
class RemoveMask(Lambda):
def __init__(self):
super(RemoveMask, self).__init__((lambda x, mask: x))
self.supports_masking = True
def compute_mask(self, input, input_mask=None):
return None
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment