- 索引号的
[0 1 2]
正数索引
0 1 2
负数索引
-3 -2 -1
a[-1:]指-1元素往后
-
隔行采样
a[: , : , 0:28:2 , 0:28:2 ] : 2的意思是0~28隔一个采样,0和28可以省去 -
index_select(dim, index)
第二个参数得是tensor类型，不能是list类型 -
ge( x):大于等于x的数置为1
-
masked_select(x,mask)
x=torch.randn(3,4)
mask=x.ge(0.5) #将大于0.5的置1，其他置0
torch.masked_select(x,mask) #就可以得到x中大于0.5的元素
- take(src,index):先把src打平，再按index取数