1. 程式人生 > >TensorFlow計算圖視覺化

TensorFlow計算圖視覺化

# -*- coding: utf-8 -*-

import tensorflow as tf
a=tf.constant(5,name='input_a')
b=tf.constant(3,name='input_b')
c=tf.multiply(a,b,name='maltiply_c')
d=tf.add(a,b,name='add_d')
e=tf.add(d,c,name='add_e')
sess=tf.Session()
sess.run(e)
#加上這行程式碼,就可以視覺化。指定檔案的路徑,沒有會自動生成。指定資料來源:計算圖
#將計算圖上面的資料寫入events檔案裡面,tensorboard可以同時從events檔案裡面讀取資料並顯示在web端
 writer=tf.summary.FileWriter('./improved_graph2',sess.graph)#tf.summary.FileWriter返回一個FileWriter類,它有很多其他方法
    writer.flush()
    writer.close()

    sess.close()

============================code=end=======

我們程式碼裡面指定了檔案的相對與當前程式碼的同級目錄下,會生成一個events.out.xx的檔案。

在終端進入該檔案路徑下輸入



然後再IE上輸入你遠端主機的 IP地址:6009.就可以看到如下