[英]Tensorflow reshape on convolution output gives TypeError
當我嘗試使用tf.reshape()
重新tf.reshape()
卷積的輸出時,我得到一個TypeError
TypeError: Expected binary or unicode string, got -1
我寫的模型是:
with tf.name_scope('conv1'):
filter = tf.Variable(tf.truncated_normal([5, 5, 1, self.num_hidden / 2], mean=0.0,
stddev=0.02, dtype=tf.float32),
name='filter')
b = tf.Variable(tf.zeros([self.num_hidden / 2], dtype=tf.float32),
name='b')
h1 = tf.nn.relu(tf.nn.bias_add(tf.nn.conv2d(inp, filter,
[1, 2, 2, 1], padding='SAME'), b))
with tf.name_scope('conv2'):
filter = tf.Variable(tf.truncated_normal([5, 5, self.num_hidden / 2, self.num_hidden], mean=0.0,
stddev=0.02, dtype=tf.float32),
name='filter')
b = tf.Variable(tf.zeros([self.num_hidden], dtype=tf.float32),
name='b')
h2 = tf.nn.relu(tf.nn.bias_add(tf.nn.conv2d(h1, filter,
[1, 2, 2, 1], padding='SAME'), b))
# h2 -> [-1, 7, 7, 32]
# num_units -> [-1, 1568]
shape = h2.get_shape()
num_units = shape[1]*shape[2]*shape[3]
with tf.name_scope('reshape'):
h2_flattened = tf.reshape(h2, [-1, num_units])
h2_flattened = tf.nn.dropout(h2_flattened, keep_prob=0.9)
with tf.name_scope('prediction'):
W = tf.Variable(tf.truncated_normal([num_units, 1], mean=0.0, stddev=0.01,
dtype=tf.float32), name='W')
b = tf.Variable(tf.zeros([1], dtype=tf.float32), name='b')
self.pred = tf.matmul(h2_flattened, W) + b
而我得到的確切錯誤是:
Traceback (most recent call last):
File "single_model_conv.py", line 108, in <module>
gan = GAN(num_latent, 28, 'single')
File "single_model_conv.py", line 23, in __init__
self.adversary(self.gen_image)
File "single_model_conv.py", line 93, in adversary
h2_flattened = tf.reshape(h2, [-1, num_units])
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/ops/gen_array_ops.py", line 1977, in reshape
name=name)
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 490, in apply_op
preferred_dtype=default_dtype)
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 657, in convert_to_tensor
ret = conversion_func(value, dtype=dtype, name=name, as_ref=as_ref)
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.py", line 180, in _constant_tensor_conversion_function
return constant(v, dtype=dtype, name=name)
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.py", line 163, in constant
tensor_util.make_tensor_proto(value, dtype=dtype, shape=shape))
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/tensor_util.py", line 422, in make_tensor_proto
tensor_proto.string_val.extend([compat.as_bytes(x) for x in proto_values])
File "/nfs/nemo/u3/idurugkar/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/util/compat.py", line 64, in as_bytes
(bytes_or_text,))
TypeError: Expected binary or unicode string, got -1
我不明白為什么會這樣。 似乎在將形狀數組轉換為張量時存在一些錯誤,但是當我嘗試將任意數組轉換為張量時,它可以正常工作。 我也嘗試將所有維度轉換為實際值(batch_size而不是-1),這也不起作用。
我的tensorflow版本是0.11,我在支持GPU的Linux機器上運行它。
我以前必須這樣做。 改變這個
shape = h2.get_shape()
對此:
shape = h2.get_shape().as_list()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.