python中[...,1]與[...,1:2]

數組的調用中經常會用到a[...,1]與a[...,1:2]最開始對這兩種調用方法有點不是很明白，操作後如下：

 1 import tensorflow as tf
2 import numpy as np
3 a = np.array([[[1,2,21],[3,4,34]],[[5,6,56],[7,8,78]]])
4 print('a.shape:',a.shape)
5
6 b = a[...,0:2]
7 print('b :',b)
8 print('shape.b:',b.shape)

結果如下：

a.shape: (2, 2, 3)
b : [[[1 2]
[3 4]]
[[5 6]
[7 8]]]
shape.b: (2, 2, 2)

但是如果用a[...,1],結果如下：

a.shape: (2, 2, 3)
b : [[2 4]
[6 8]]
shape.b: (2, 2)

可以看到[...,1:3]代表的時最後一維度中的1:3的數，輸入為原先維度中最後一個維度的數組，而[...,2]則是-2維度中的信息，輸入相對與原先的維度少了一個維度。