tensorflow得到张量大小但无法初始化tf.Variable size

时间:2016-05-11 20:40:01

标签: python tensorflow

我尝试使用占位符来保存列表,并获取它的长度。然后,当我用它来初始化我的变量时,它会给出错误

以下是创建错误的示例代码:

import tensorflow as tf

a = tf.placeholder(tf.int32, shape=[None])
story_mask=[4,7,10,12,15]
input_feed={}
input_feed[a.name] = story_mask
b = tf.reshape(tf.shape(a), [])
weights = tf.Variable(tf.truncated_normal([b, b], -0.1, 0.1), name="l1_weights")
c = tf.matmul(a, weights)

sess = tf.Session()
print(sess.run(c,input_feed))

错误消息位于:

weights = tf.Variable(tf.truncated_normal([b, b], -0.1, 0.1), name="l1_weights")

和消息是:

  

TypeError:预期单个Tensor时的张量列表

任何人都知道我错在哪里。谢谢

0 个答案:

没有答案