关于tensorflow的axis的理解

Posted 2021-01-02 charlesblc

通过这篇文章，能够理解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