pytorch(六):索引与切片
一、索引
注意,如同1:,或者:2,指的是该维中,从1取到末尾。或者是从开头取到2(不包含)
如:([3,28,28]),([:2, 28, 28])指的是在第一维中,取0,1,不包含2.
如([-1:,28,28])中,由于索引是负数,因此编号成了从右往左数,从-1依次减一,索引是(-3,-2,-1)
因此,-1: 变成了-1:-1
:-1就变成了-3:-1
二、从某个维度选择具体索引
三、通过掩码选择,和通过展平后选择具体索引