[:, :, :-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开始,到最后第二个维度。如下图。

猜你喜欢

转载自blog.csdn.net/Gussss/article/details/83109860
1
(1)
>&1