TensorFlow:值错误形状和等级不匹配: ValueError:形状(?，128,128，2)必须具有等级2

Question

当我使用deconv神经网络运行预测时，我得到了这个异常。排名和形状似乎是一样的，所以我不确定是什么问题。

File "/home/Workspace/image-recognition/app/model/per_pixel_deconv.py", line 141, in Model
    softmax = tf.nn.softmax(output, name=None)

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/ops/gen_nn_ops.py", line 1396, in softmax
    result = _op_def_lib.apply_op("Softmax", logits=logits, name=name)

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 703, in apply_op
    op_def=op_def)

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 2319, in create_op
    set_shapes_for_outputs(ret)

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1711, in set_shapes_for_outputs
    shapes = shape_func(op)

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/framework/common_shapes.py", line 45, in _ShapeFunction
    return [op.inputs[0].get_shape().with_rank(rank)]

File "/home/anaconda2/envs/image-recognition/lib/python2.7/site-packages/tensorflow/python/framework/tensor_shape.py", line 641, in with_rank
    raise ValueError("Shape %s must have rank %d" % (self, rank))
ValueError: Shape (?, 128, 128, 2) must have rank 2

Answer 1

Shape (?, 128, 128, 2)是一个秩4的张量，但是softmax算子显然需要一个秩2的张量。

在我看来，您使用的是旧版本的Tensorflow。在版本0.10中，tf.nn.softmax要求其输入等级为2：https://www.tensorflow.org/versions/r0.10/api_docs/python/nn/classification#softmax

在当前版本的Tensorflow中，tf.nn.softmax接受任何等级的输入：https://www.tensorflow.org/api_docs/python/nn/classification#softmax

尝试使用较新版本的Tensorflow。

希望这能有所帮助！