Tensorflow Matmul尺寸错误

Question

我正在运行以下代码，最后一行引发错误。

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)

错误：

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

我对为什么会收到此错误感到困惑。 [96,366]矩阵乘以[366,128]矩阵是否有效？

谢谢

Answer 1

得到它了。 这是我代码中的错字（说366而不是336）。