python中slice用法

slice() 函数返回 slice 对象（切片）。

slice 对象用于指定如何对序列进行裁切。您可以指定在哪里开始裁切以及在哪里结束裁切。您还可以指定步进，例如只切每隔一个项目。

语法

slice(start, end, step)

默认为step为1

x = slice(2,6)
print(x)

#输出：  slice(2, 6, None)

y = torch.randn(6,7)
print(y)
print(y[x][:])
print(y[x,x])

输出：y[x]输出的是y[2:6]，y[x,x]输出的是y[2:6,2:6]

下面是指定步长的做法：

z = slice(0,6,2)
print(y[z])

输出的是y[0,2,4]