关于python中的[:, -1, :] [:, :, -1] [-1, :, :]

其实是个比较简单的知识点，但是因为经常忘记，在此记录一下

以[:, -1, :]为例

冒号代表该维度的全部元素

-1代表该维度的最后一个元素

如

output = np.array([i for i in range(8)]).reshape(2, 2, 2)
print(output)

输出的是

[[[0 1]
  [2 3]]

 [[4 5]
  [6 7]]]

而经过[:, -1, :]处理后

print(output[:, -1, :])

变成

[[2 3]
 [6 7]]

[2 3]是第二维中最后一个元素的全部内容

注意这个结果其实降低了一个维度，结果是一个二维矩阵

同样的我们可以看一下[:, :, -1]和[-1, :, :]的结果

print(output[:, :, -1])
print(output[-1, :, :])

分别是

[[1 3]
 [5 7]]


[[4 5]
 [6 7]]

结果其实降低了一个维度，是一个二维矩阵