使用tf.print()打印tensor內容


使用tf.Print()打印tensor內容,這是tensorflow中調試bug的一個手段,例子如下所示:

import tensorflow as tf

a = tf.Variable(tf.random_normal([3, 3, 1, 64], stddev=0.1))
a = tf.Print(a, [a], "a: ",summarize=9)
init = tf.global_variables_initializer()
sess = tf.Session()
sess.run(init)
sess.run(a)

輸出結果

tf.Print()函數參數為:

Print(
    input_,
    data,
    message=None,
    first_n=None,
    summarize=None,
    name=None
)

參數:

  • input_:通過這個操作的張量。
  • data:計算 op 時要打印的張量列表。
  • message:一個字符串,錯誤消息的前綴。
  • first_n:只記錄 first_n 次數。負數日志,這是默認的。 
  • summarize:只打印每個張量的許多條目。如果沒有,則每個輸入張量最多打印3個元素。
  • name:操作的名稱(可選)。

返回:

該操作將返回與 input_ 相同的張量。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM