直接给代码
import tensorflow as tf
x = tf.placeholder(shape=tf.TensorShape([]), dtype=tf.float32)
y = tf.placeholder(shape=tf.TensorShape([None]), dtype=tf.float32)
sess = tf.Session()
sess.run(tf.global_variables_initializer())
print(sess.run([x, y], feed_dict={x: 5.0, y:[5.0, 5.0, 5.0]}))
输出结果:
[array(5., dtype=float32), array([5., 5., 5.], dtype=float32)]
tf.TensorShape([])表示:只能给变量提供一个数字
tf.TensorShape([None])表示:只能给变量提供一个列表,长度未知的向量