tensorflow multiplier deux tenseurs

Question

Je suis en train de multiplier deux tenseurs ensemble que les deux ont la même forme:

weights = tf.Variable(tf.random_normal([200], stddev=0.35), 
         name="weights") 
weights2 = tf.Variable(tf.random_normal([200], stddev=0.35), 
         name="weights2") 


greg = tf.matmul(weights,weights2) 


sess=tf.Session() 
sess.run(tf.initialize_all_variables()) 
sess.close() 

Un essai en bloc-notes jupyter, je reçois cette erreur:

« Formes (200,) et (?,?) doivent avoir le même rang "

Qu'est-ce qui me manque?

Answer 1

Comme NPE mentionne dans their comment, le tf.matmul() op attend ses deux entrées à tenseurs en deux dimensions, mais vos arguments weights et weights2 sont unidimensionnels tenseurs.

Si vous voulez calculer le produit intérieur de ces deux tenseurs, vous devez les remodeler à 200 par 1 et 1 par 200 matrices, en utilisant (par exemple) tf.reshape() comme suit:

greg = tf.matmul(tf.reshape(weights, [1, 200]), tf.reshape(weights2, [200, 1]))