TF 错误：InvalidArgumentError (see above for traceback): Reduction axis 0 is empty in shape [ ]

Winter-Solider

于 2018-06-21 10:08:16 发布

阅读量3.5k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_22910255/article/details/80756198

本文介绍了一个关于TensorFlow中使用argmax函数时出现的InvalidArgumentError错误，并通过一个具体的例子解释了如何重现该错误及原因。当输入张量为空时，argmax会抛出异常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Tensorflow.python.framework.errors_impl.InvalidArgumentError: Reduction axis 0 is empty in shape [0,100]

[[Node: ArgMax = ArgMax[T=DT_DOUBLE, Tidx=DT_INT32, output_type=DT_INT64, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Const, ArgMax/dimension)]]

原因：因为tf.argmax（）中的数组是个空数组

a = tf.constant(np.zeros([0,100]))
b = tf.argmax(a, axis=0)
with tf.Session() as sess:
    print(b)
    #如果把下面一行屏蔽打开，就会报错
    # print(sess.run(b))