在TensorFlow API中创建常量的函数原型如下所示:
tf.constant(
value,
dtype=None,
shape=None,
name='Const',
verify_shape=False
)
可以看到第一个值value是必须的,可以是一个数值,也可以是一个列表。
如果是数值:
tensor=tf.constant(1)
为查看结果必须创建一个会话,并用取值函数eval()来查看创建的tensor的值:
sess=tf.Session()
with sess.as_default():
print('结果是:', tensor.eval())
结果是:1
而如果value是一个列表:
tensor=tf.constant([1, 2])
sess=tf.Session()
with sess.as_default():
print('结果是:', tensor.eval())
结果是:[1 2]
后面四个参数可写可不写,第二个参数表示数据类型,一般可以是tf.float32, tf.float64等:
tensor=tf.constant([1, 2], dtype=tf.float32)
sess=tf.Session()
with sess.as_default():
print('结果是:', tensor.ev