选取部分张量元素

代码1：

t=torch.arange(12)
print(t)
print(t[3],t[0])
print(t[-5])
print(t[3:6])#个数为3
print(t[:6])
print(t[-5:])
print(t[3:6:2])#2表示只有2个元素
print(t[3::2])#以2位方差

 运行结果：

代码2 ：

 

t= torch.arange(24).reshape(2,3,4)#大小（2,3,4）
print(t)
print(t[1:,-2])
print(t[1:,-2,2])
print(t[0,:])

运行结果：

 

 

代码3：

t= torch.arange(24).reshape(2,3,4)#大小（2,3,4）
print(t)
index=torch.tensor([2,2])#表示对那个维度进行处理
print(t.index_select(1,index))

运行结果