Tensorflow matmul尺寸误差

时间:2017-09-20 18:59:41

标签: python machine-learning tensorflow neural-network

我正在运行以下代码,最后一行会抛出错误。

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]矩阵工作?

由于

1 个答案:

答案 0 :(得分:0)

知道了。这是我的代码中的拼写错误(366而不是336)。