维度和axis

1 如何看shape:看括号

array([[[[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]],

        [[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]]]])
#[1,2,3,4]

从[[[[可知是四维向量,第一个 [ 不看,第二个[ 括号的个数为1,第三个[ 括号的个数为2,第四个[ 括号的个数为3,括号里有四个数。故shape为[1,2,3,4]。

2如何看axis:看shape,看括号

shape = (    1,    2,     3,     4     )
             |      |     |      |
axis  =      0,     1,    2,   3==-1
a=array([[[1],
          [9],
          [5]],

         [[5],
          [7],
          [1]]])

#b=a[1,:,:] 即取axis=0的第1+1个维度
b=array([[[5],
          [7],
          [1]]])

#c=a[:,:,2] 即取axis=2的第2+1个维度
不存在
a=array([[[2, 7],
          [9, 3],
          [9, 5]],

         [[1, 9],
          [2, 9],
          [5, 5]]])

#b=a[:,:,1] 即取axis=2的第1+1个维度

b=array([[7,3,5],
         [9,9,5]])

拓展维度: tf.expand_dims()_GungnirsPledge的博客-CSDN博客_tf 扩展维度

猜你喜欢

转载自blog.csdn.net/Fwuyi/article/details/125746194