Placeholder 可以指定參數化來接受外部輸入。
a = tf.placeholder(tf.float32)
b = tf.placeholder(tf.float32)
adder_node = a + b # 函數的設定,也可以寫成 adder_node = tf.add(a, b)
前面的三行有點像函數的設定,其中定義兩個輸入參數(a和b),然後對它們進行操作。 我們可以通過使用run方法的將參數輸入具體值。
範例程式05_placeholder.py
- import tensorflow as tf
- sess = tf.Session()
- a = tf.placeholder(tf.float32) # 變數a 為 float32
- b = tf.placeholder(tf.float32) # 變數b 為 float32
- adder_node = tf.add(a, b) # 定義相加函數
- adder_node = a + b # 也可以這樣寫
- print(sess.run(adder_node, {a: 3, b: 4.5})) # 結果 7.5
- print(sess.run(adder_node, {a: [1, 3], b: [2, 4]})) # 結果[ 3. 7.]
圖6 執行結果