- 博客(2)
- 收藏
- 关注
RSS订阅原创 Pytorch中[2::3]的使用
示例1import torcha = torch.rand((9, ))print(a)b = a[2::3]print(b)结果：可以看出b是的元素是从a[2]开始，逐次增加3，b = [a[2], a[5], a[8]]示例2import torcha = torch.rand((10, ))print(a)b = a[2::3]print(b)结果：可以看出如果下标超出了a的范围，则舍弃，a[11]不出现在b中示例3import torcha = .
2020-07-27 23:27:49
954
原创 PyTorch中[..., 0]的使用
1. 示例1import torcha = torch.rand((17, 24, 8))b = a[..., 0]c = a[:, :, 0]print(b == c)结果为True2. 示例2import torcha = torch.rand((64, 17, 24, 8))b = a[..., 0]c = a[:, :, :, 0]print(b == c)结果为True3. 结论可以看出[…, 0]相当于[:, :, … :, 0]...
2020-07-27 23:05:49
1055
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人