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

第十行返回的是一个标量。
以上是正常的索引,还有一些高级的索引:
比如我要取两张:

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

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

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

第一行你会发现有两个冒号这种东西,就意味着要隔行采样了;
TIPS:把冒号理解成箭头;
![]()
以上是索引的功能,切片就不一样在,我要的是具体的哪一块:

还有一个神奇的符号:

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

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

take函数也是如此:

2281

被折叠的 条评论
为什么被折叠?



