Enforcing dtype in K.zeros_like for mask

Enforcing dtype in `K.zeros_like` for mask in `recursion` function. `K.zeros_like` failed to infer dtype of `mask` when a Masking layer is used.
ashutoshsingh0223 · Sep 12, 2019 · 0d4129f · 0d4129f
1 parent 5ffab17
commit 0d4129f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_contrib/layers/crf.py b/keras_contrib/layers/crf.py
@@ -513,7 +513,7 @@ def recursion(self, input_energy, mask=None, go_backwards=False,
         constants = [chain_energy]
 
         if mask is not None:
-            mask2 = K.cast(K.concatenate([mask, K.zeros_like(mask[:, :1])], axis=1),
+            mask2 = K.cast(K.concatenate([mask, K.cast(K.zeros_like(mask[:, :1]), mask.dtype)], axis=1),
                            K.floatx())
             constants.append(mask2)