1. 程式人生 > >TensorFlow函式之tf.constant()

TensorFlow函式之tf.constant()

tf.constant()可以實現生成一個常量數值。


tf.constant()格式為:

tf.constant(value,dtype,shape,name)

引數說明:

  1. value:常量值
  2. dtype:資料型別
  3. shape:表示生成常量數的維度
  4. name:資料名稱

下邊生成一個0.1常量大小為2*2的資料:

import tensorflow as tf

v = tf.constant(0.1, dtype=tf.float32, shape=[2, 2], name='v')

sess = tf.Session()
print(sess.run(v))
sess.close()

輸出為:

[[0.1 0.1]
 [0.1 0.1]]