Quick fix

ShieLian · web-flow · commit 7ddf0769a1c2 · 2017-07-17T21:59:36.000+08:00
diff --git a/tensorlayer/layers.py b/tensorlayer/layers.py
@@ -2174,7 +2174,7 @@ def _PS(X, r, n_out_channel):
             bsize = tf.shape(X)[0] # Handling Dimension(None) type for undefined batch dim
             Xs=tf.split(X,r,3) #b*h*w*r*r
             Xr=tf.concat(Xs,2) #b*h*(r*w)*r
-            X=tf.reshape(Xr,(b,r*a,r*b,c)) # b*(r*h)*(r*w)*c
+            X=tf.reshape(Xr,(bsize,r*a,r*b,c)) # b*(r*h)*(r*w)*c
         else:
             print(_err_log)
         return X
