Embedding layer - cast float to int implicitly (keras-team#3342)

* Embedding layer - cast float to int implicitly * Cast only if not int
tushuhei · Jul 29, 2016 · cb4de1f · cb4de1f
1 parent b6d23b2
commit cb4de1f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras/layers/embeddings.py b/keras/layers/embeddings.py
@@ -125,6 +125,8 @@ def get_output_shape_for(self, input_shape):
         return (input_shape[0], input_length, self.output_dim)
 
     def call(self, x, mask=None):
+        if K.dtype(x) != 'int32':
+            x = K.cast(x, 'int32')
         if 0. < self.dropout < 1.:
             retain_p = 1. - self.dropout
             B = K.random_binomial((self.input_dim,), p=retain_p) * (1. / retain_p)