1. 程式人生 > 程式設計 >解決tensorflow列印tensor有省略號的問題

解決tensorflow列印tensor有省略號的問題

先上程式碼:

import tensorflow as tf
x = tf.ones(shape=[100,200],dtype=tf.int32,name='x')
y = tf.zeros(shape=[2,3],dtype=tf.float32,name='y')
with tf.Session() as sess:
  print(sess.run([x,y]))

輸出結果如下:

解決tensorflow列印tensor有省略號的問題

如果我除錯的時候想檢視省略號代表的值是什麼

只需要改成如下程式碼就行:

import tensorflow as tf
import numpy as np #藉助numpy模組的set_printoptions()函式,將列印上限設定為無限即可
np.set_printoptions(threshold=np.inf)
x = tf.ones(shape=[10,20],name='y')
with tf.Session() as sess:
  print(sess.run(x,y))

輸出結果如下:可以看到省略號的部分也都打出來了

解決tensorflow列印tensor有省略號的問題

以上這篇解決tensorflow列印tensor有省略號的問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。