Add elemwise param to get_constant_idx

rasmi · Sep 9, 2016 · e7fe132 · e7fe132
1 parent e77d567
commit e7fe132
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/theano/tensor/subtensor.py b/theano/tensor/subtensor.py
@@ -398,7 +398,7 @@ def convert(entry, slice_ok=True):
             raise AdvancedIndexingError(Subtensor.e_indextype, entry)
 
     def get_constant_idx(self, inputs, allow_partial=False,
-                         only_process_constants=False):
+                         only_process_constants=False, elemwise=True):
         """
         Return the idx_list with constant inputs replaced by their
         python scalar equivalent.
@@ -442,7 +442,8 @@ def conv(val):
                 try:
                     return get_scalar_constant_value(
                         val,
-                        only_process_constants=only_process_constants)
+                        only_process_constants=only_process_constants,
+                        elemwise=elemwise)
                 except theano.tensor.NotScalarConstantError:
                     if allow_partial:
                         return val