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]

掩码选取

打平选取