axis=-1,0,1的含义

axis=-1,0,1的含义

axis的本意是轴的意思。
在python中,axis代表的是多维数组中数据操作的方向。
举例说明,在pycharm环境创建一个三维数组:
import numpy as np
import tensorflow as tf
b = np.arange(27).reshape(3,3,3)
print(b)
a = np.max(b, axis=-1)
print(“a=”,a)
运行结果:
[[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]

[[ 9 10 11]
[12 13 14]
[15 16 17]]

[[18 19 20]
[21 22 23]
[24 25 26]]]

a= [[ 2 5 8]
[11 14 17]
[20 23 26]]
如果把a = np.max(b, axis=-1) 改成a = np.max(b, axis=0),则运行结果如下:
[[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]

[[ 9 10 11]
[12 13 14]
[15 16 17]]

[[18 19 20]
[21 22 23]
[24 25 26]]]

a= [[18 19 20]
[21 22 23]
[24 25 26]]
如果把a = np.max(b, axis=-1) 改成a = np.max(b, axis=1),则运行结果如下:
[[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]

[[ 9 10 11]
[12 13 14]
[15 16 17]]

[[18 19 20]
[21 22 23]
[24 25 26]]]

a= [[ 6 7 8]
[15 16 17]
[24 25 26]]

发布了4 篇原创文章 · 获赞 0 · 访问量 176

猜你喜欢

转载自blog.csdn.net/sinat_38819901/article/details/104855189