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))
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)