tensorflow mnist read_data_sets fails

下载处理mnist数据时出现如下错误

VisibleDeprecationWarning: converting an array with ndim > 0 to an index will result in an error in the future

解决方法：

在input_data.py文件中return numpy.frombuffer(bytestream.read(4), dtype=dt) 后添加[0]

 

return numpy.frombuffer(bytestream.read(4), dtype=dt)[0]