通过这篇文章，能够理解tensorflow里面的axis：
https://blog.csdn.net/fangjian1204/article/details/53055219
可以总结为一句话：设axis=i，则numpy沿着第i个下标变化的放下进行操作。
理解一下：
>>> import numpy as np >>> data = np.array([ ... [1,2,1], ... [0,3,1], ... [2,1,4], ... [1,3,1]]) >>> np.sum(data, axis=1) array([4, 4, 7, 5]) >>> np.min(data, axis=0) array([0, 1, 1])
另外还有：
>>> np.average(data)
1.6666666666666667
原文：https://www.cnblogs.com/charlesblc/p/9555652.html