fix dgcnn mask is none

xv44586 · Oct 12, 2020 · 4ad480d · 4ad480d
1 parent a638f82
commit 4ad480d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/toolkit4nlp/layers.py b/toolkit4nlp/layers.py
@@ -486,6 +486,9 @@ def call(self, x, mask=None):
         if self.dropout_rate is not None:
             g = K.in_train_phase(K.dropout(g, self.dropout_rate), g)
         g = K.sigmoid(g)
+        # mask is none
+        mask = mask if mask is not None else K.ones_like(x)
+
         if self.skip_connection:
             if K.int_shape(x0)[-1] != self.o_dim:
                 x0 = self.conv1d_1x1(x0)