江湖小Jay的博客

np.argmax(a, axis=None, out=None)tf.argmax(input, axis=None, name=None, dimension=None, output_type=dtypes.int64):np.argmax()与tf.agrmax()函数用法类似，用于寻找每一行或者每一列中的最大值的索引值，axis的值代表行或列，分别表示为axis=0 按列寻找、axis=1 按行寻找.

安装GPU版本的TensorFlow，首先需要VS环境，如果没有需要安装，但是VS全部安装会占内存，因此可以去Download Visual C++ Redistributable for Visual Studio 2015 from Official Microsoft Download Center对应下载安装vc_redist2015.x64.exe 和vc_redist2015.x86.exe即可安装好VS环境后，再安装CUDA安装CUDA前需要检查自己电脑的NVIDIA版本，然后查找