slice

python 内建函数

class slice(start, end, step)
class slice(stop)
arg:

• start – 起始位置
• stop – 结束位置
• step – 间距
  实现切片对象，通常我们切片可以直接[:1:2:]这种方式进行，但是有时候切片对象是一个变量，此时就用到了slice

demo：

s1  = slice(0,-1,1)  #设置后，放在某个维度上
y = torch.arange(9).view(3,3)
print(y)
print(y[s1,:]==y[0:-1,:]) #行
print(y[:,s1]==y[:,0:-1]) #列

输出

tensor([[0, 1, 2],
        [3, 4, 5],
        [6, 7, 8]])
tensor([[True, True, True],
        [True, True, True]])
tensor([[True, True],
        [True, True],
        [True, True]])