2017-07-17 4 views
0

Ich habe einen einfachen Code geschrieben, um die Tensorflow summarize-Funktion auszuprobieren. Der Code ist unten.InvalidArgumentError bei der Verwendung der Zusammenfassung in Tensorflow v1.2.1

import tensorflow as tf 
import numpy as np 

graph = tf.Graph() 


with graph.as_default(): 
    x = tf.placeholder(tf.float32, [1, 2], name='x') 
    W = tf.ones([2, 1], tf.float32, name='W') 
    b = tf.constant([1.5], dtype=tf.float32, shape=(1, 1), name='bias') 
    y_ = tf.add(tf.matmul(x, W, name='mul'), b, name='add') 
tf.summary.scalar('y', y_) 

with tf.Session(graph=graph) as session: 
    merged = tf.summary.merge_all() 
    fw = tf.summary.FileWriter("/tmp/tensorflow/logs", graph=graph) 

    tf.global_variables_initializer().run() 
    x_var = np.array([1., 1.], np.float32).reshape([1, 2]) 
    print(x_var) 
    summary, y = session.run([merged, y_], feed_dict={x: x_var}) 
    fw.add_summary(summary, 0) 
    print(y) 

    fw.close() 

Grundsätzlich versucht es y=Wx + b zu implementieren.

Der Code funktioniert, wenn ich alle Zusammenfassung verwandten Code entfernen. Aber wenn ich den Zusammenfassung zugehörigen Code hinzufügen, ich habe unten Fehler:

InvalidArgumentError (see above for traceback): tags and values not the same shape: [] != [1,1] (tag 'y') 
    [[Node: y = ScalarSummary[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/cpu:0"](y/tags, add)]] 

ich sowohl in der normalen Python versucht, und IPython.

Antwort

0

Tags und Werte haben nicht die gleiche Form. Sie übergeben x_var, die ein Vektor ist und summary einen Skalarwert verwendet. Sie können einfach tf.reduce_mean verwenden, dieses Problem zu lösen:

with graph.as_default(): 
    x = tf.placeholder(tf.float32, [None, 2], name='x') 
    W = tf.ones([2, 1], tf.float32, name='W') 
    b = tf.constant([1.5], dtype=tf.float32, shape=(1, 1), name='bias') 
    y_ = tf.add(tf.matmul(x, W, name='mul'), b, name='add') 
    tf.summary.scalar('y', tf.reduce_mean(y_)) 

Dies wird einen skalaren Wert erstellen.

+0

Mit Hilfe Ihrer Antwort und [diese] (http://www.cloudypoint.com/Tutorials/discussion/python-solved-tensorflow-how-to-tensorboard/) Post, bekomme ich endlich den Code arbeiten. Ich habe deine Antwort auch aktualisiert. – davidshen84

+0

Die ** Schlüssel ** sind, Eingabe ** x ** sollte NICHT feste Größe sein; Verwenden Sie * reduzieren _ ** auf die Ausgabe ** y **. – davidshen84

Verwandte Themen