diff options
| -rw-r--r-- | neural_style.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/neural_style.py b/neural_style.py index b220f12..13ba513 100644 --- a/neural_style.py +++ b/neural_style.py @@ -382,8 +382,8 @@ def mask_style_layer(a, x, mask_img): mask = tf.stack(tensors, axis=2) mask = tf.stack(mask, axis=0) mask = tf.expand_dims(mask, 0) - a = tf.mul(a, mask) - x = tf.mul(x, mask) + a = tf.multiply(a, mask) + x = tf.multiply(x, mask) return a, x def sum_masked_style_losses(sess, net, style_imgs): |