我在Tensorflow中实现了一个ConvNet:具有3个Conv-Relu-Pool层,一个FC层和5个softmax分类器。 在图表中(在训练时),我将所有5个softmax连成一个张量:
train_prediction = tf.pack([tf.nn.softmax(logit_d1),\
tf.nn.softmax(logit_d2), tf.nn.softmax(logit_d3),\
tf.nn.softmax(logit_d4),tf.nn.softmax(logit_d5)],\
0, name="tf_train_prediction")
当我运行代码时,我收到标题中显示的错误。
在图表中,我还有2个tf.pack()
操作:一个名为" validation_prediction
",一个没有名字。此外,在代码中,我没有tf_train_prediction
的重复名称。
我不明白为什么我会得到那个错误信息。如果您有任何建议,请告诉我。仅供参考,我正在运行Tensorflow版本0.11
为了完整性,以下是其他2个tf.pack()操作:
valid_prediction = tf.pack(\
[tf.nn.softmax(val_logit_d1),tf.nn.softmax(val_logit_d2),\
tf.nn.softmax(val_logit_d3),tf.nn.softmax(val_logit_d4),\
tf.nn.softmax(val_logit_d5)], 0, name="valid_prediction")
predicted_digits = tf.transpose( tf.pack(\
[tf.argmax(softmax_d1[:,:], 1),tf.argmax(softmax_d2[:,:],1),\
tf.argmax(softmax_d3[:,:], 1),tf.argmax(softmax_d4[:,:],1),\
tf.argmax(softmax_d5[:,:], 1)]) )