diff options
Diffstat (limited to 'neural_style.py')
| -rw-r--r-- | neural_style.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/neural_style.py b/neural_style.py index d0bb45f..1dd0c6c 100644 --- a/neural_style.py +++ b/neural_style.py @@ -358,7 +358,7 @@ def style_layer_loss(a, x): return loss def gram_matrix(x, area, depth): - F = tf.reshape(x[0], (area, depth)) + F = tf.reshape(x, (area, depth)) G = tf.matmul(tf.transpose(F), F) return G |