diff options
| author | Cameron <cysmith1010@gmail.com> | 2017-01-05 18:26:05 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-01-05 18:26:05 -0700 |
| commit | bc216125c60b902b767bc3ccd97fb49bb3b55580 (patch) | |
| tree | 1c34cdc4a14f288d7bd81d3c530f5ccffd39d4d3 /neural_style.py | |
| parent | af50ae73cfe6f1533e5390e5131e25c6d322dc8a (diff) | |
| parent | 8e444ab2ad4dcbec48c1fe775c4512fa4ed390a3 (diff) | |
Merge pull request #19 from ProGamerGov/patch-1
Replaced outdated tf.pack with tf.stack
Diffstat (limited to 'neural_style.py')
| -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 8d40e1b..be5a95e 100644 --- a/neural_style.py +++ b/neural_style.py @@ -380,8 +380,8 @@ def mask_style_layer(a, x, mask_img): tensors = [] for _ in range(d.value): tensors.append(mask) - mask = tf.pack(tensors, axis=2) - mask = tf.pack(mask, axis=0) + 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) |