From 5ca01d831a0ceec853135b9eb35e65ed8e327d82 Mon Sep 17 00:00:00 2001
From: cam <cameron@ideum.com>
Date: Sat, 15 Oct 2016 16:36:24 -0600
Subject: Fixed bug between tensorflow builds 0.9, 0.10rc, 0.10

---
 neural_style.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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
 
-- 
cgit v1.2.3-70-g09d2