pytorch实践学习---4.索引与切片

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-21 14:05   22   0

首先我们来看PyTorch的索引问题,就和python语言一样,正常能用脑子想出来的逻辑:

第十行返回的是一个标量。

以上是正常的索引,还有一些高级的索引:

比如我要取两张:

还有检验两张图片的第一个通道上的索引:(常识:一共有三个通道:RGB)

还有-1的情况,就是倒着来:



遵循这样的原则:

接下来,我不想连续的取了,我要跳着取,就是这样:

第一行你会发现有两个冒号这种东西,就意味着要隔行采样了;

TIPS:把冒号理解成箭头;

以上是索引的功能,切片就不一样在,我要的是具体的哪一块:

还有一个神奇的符号:

三个点就代表,所有的维度我都要;

还有一个不常用的切片,因为mask 会把数据打平:

take函数也是如此:

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP