版权声明:本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明出处! https://blog.csdn.net/huplion/article/details/80340310
这个bug发生在 读取TFRecord格式的文件中。
解决方案是tf.FixedLenFeature后面加上具体维度如:
features = tf.parse_single_example(serialized_example,
features={
'label': tf.FixedLenFeature([381], tf.int64)
})