超级会员免费看
本文重点
有时候当我们拥有一个Tensor张量的时候，我们可能需要获取它某一维度的信息，那么此时我们就需要索引和切片的技术，它们可以帮助我们解决这些问题。
切片操作
a是四维的，然后默认是从第一维开始取，逗号表示取不同的维度
a[:2]表示第一维取0，1，后面三维取所有
a[:2,:2]表示第一维取0，1，第二维取0，1，后面两维取所有
前两维取所有，第三维和第四维间隔取样，0到28，每两维取样一次
0：28：2等于：：2
索引下标
通过下标进行索引
每天五分钟深度学习PyTorch：Tensor张量的索引和切片
最新推荐文章于 2024-07-31 20:39:38 发布
本文介绍了在PyTorch中如何对Tensor进行索引和切片操作，包括切片的基本用法，如a[:2]和a[:2,:2]，以及不同类型的索引方式，如index_select、take和masked_select。这些技术对于处理和提取Tensor特定维度的信息至关重要。"
80963858,7778221,Python合格率计算实践,"['Python', '编程练习', '数据处理']
订阅专栏 解锁全文
扫一扫
3568
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
