Merge 06d9faf into ba16a2a

Bihaqo · Jun 7, 2018 · fad033b · fad033b
2 parents ba16a2a + 06d9faf
commit fad033b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/t3f/regularizers.py b/t3f/regularizers.py
@@ -34,7 +34,7 @@ def l2(tt):
       my_scale = tf.convert_to_tensor(scale,
                                        dtype=tt.dtype.base_dtype,
                                        name='scale')
-      return tf.mul(my_scale, ops.frobenius_norm_squared(tt), name=name)
+      return tf.multiply(my_scale, ops.frobenius_norm_squared(tt), name=name)
 
   return l2
 
@@ -74,6 +74,6 @@ def regularizer(tt):
       penalty = 0.0
       for i in range(tt.ndims()):
         penalty += core_regularizer(tt.tt_cores[i])
-      return tf.mul(my_scale, penalty, name=name)
+      return tf.multiply(my_scale, penalty, name=name)
 
   return regularizer