Tensorflow matmul dimensions erreur

Question

Je cours le code suivant et la ligne finale jette une erreur.

weights = tf.Variable(
tf.truncated_normal([DATA_POINTS, hidden1_units], 
         stddev=1.0/math.sqrt(float(DATA_POINTS))), 
    name='weights') 
biases = tf.Variable(tf.zeros([hidden1_units]), 
        name='biases') 
hidden1 = tf.nn.relu(tf.matmul(images, weights) + biases) 

Erreur:

InvalidArgumentError (see above for traceback): Matrix size-incompatible: In[0]: [96,336], In[1]: [366,128] 

Je suis confus pourquoi je reçois cette erreur. Ne devrait-on pas multiplier une matrice [96,366] par une oeuvre matricielle [366,128]?

Merci

Answer 1

Vous l'avez. C'était une faute de frappe dans mon code (dit 366 au lieu de 336).