pytorch学习笔记3 tensor索引和切片

dim 0 占先

切片 （前N或者后N个）

：2 表示 0到2（不包含2）， 1：表示 1到末尾， -1表示最后一个元素，-2表示倒数第二个

0:28:2 表示从0到27隔点采样

：：2 表示从头到尾隔点采样

上图在0维上取0和2两个元素

...表示任意都维度（根据原数据shape和给定形式推断），且都是满选

a[0,...]其实=a[0,:,:,:]=a[0]

a[:,1,...]其实=a[:,1,:,:]=a[:,1]

掩码选取

打平选取