1. 程式人生 > >[:, :, :-1]與[:, :, -1]

[:, :, :-1]與[:, :, -1]

import numpy as np
a=np.array(range(24)).reshape(2,3,4)

anchor_state   = a[:, :, -1] 
b = a[:, :, 3]


labels         = a[:, :, :-1]
c=a[:, :, 0:3]

-1代表本維度最大的那個值,如anchor_state   = a[:, :, -1] ,相當於b = a[:, :, 3],相當於抽取a中4對應的那個維度,抽取第三個。如下圖

 一但變成labels         = a[:, :, :-1],:-1,代表從0開始,到最後第二個維度。如下圖。