- 示例1
import torch
a = torch.rand((9, ))
print(a)
b = a[2::3]
print(b)
结果:
可以看出b是的元素是从a[2]开始,逐次增加3,b = [a[2], a[5], a[8]]
- 示例2
import torch
a = torch.rand((10, ))
print(a)
b = a[2::3]
print(b)
结果:
可以看出如果下标超出了a的范围,则舍弃,a[11]不出现在b中
- 示例3
import torch
a = torch.rand((9, 2))
print(a)
b = a[2::3]
print(b)
结果:
可以看出和示例1类似b=[a[2, :], a[5, :], a[8,:]],只不过需要囊括剩下所有维的数据,更加高维的同理可得。