コード例 #1
0
 def call(self, inputs):
   boolean_mask = K.any(
       K.not_equal(inputs, self.mask_value), axis=-1, keepdims=True)
   return inputs * K.cast(boolean_mask, K.floatx())
コード例 #2
0
 def compute_mask(self, inputs, mask=None):
   return K.any(K.not_equal(inputs, self.mask_value), axis=-1)
コード例 #3
0
ファイル: core.py プロジェクト: maony/tensorflow
 def call(self, inputs):
   boolean_mask = K.any(
       K.not_equal(inputs, self.mask_value), axis=-1, keepdims=True)
   return inputs * K.cast(boolean_mask, K.floatx())
コード例 #4
0
ファイル: core.py プロジェクト: maony/tensorflow
 def compute_mask(self, inputs, mask=None):
   return K.any(K.not_equal(inputs, self.mask_value), axis=-1)
コード例 #5
0
ファイル: embeddings.py プロジェクト: AlbertXiebnu/tensorflow
 def compute_mask(self, inputs, mask=None):
   if not self.mask_zero:
     return None
   else:
     return K.not_equal(inputs, 0)
コード例 #6
0
ファイル: embeddings.py プロジェクト: woodhaha/tensorflow-1
 def compute_mask(self, inputs, mask=None):
     if not self.mask_zero:
         return None
     else:
         return K.not_equal(inputs, 0)