每天五分钟玩转深度学习框架PyTorch:选择函数where和gather

时间:2024-10-12 06:57:26

本文重点

如图表所示,这几个方法可以理解为索引函数,有些函数在切片和索引一章进行了简单的介绍,本文将再次进行介绍,温故知新。

index_select

通过特殊的索引来获取数据index_select,这个这样来理解,第一个参数表示a的第几维度,第二个参数表示获取该维度的哪部分。

我们把16,3,28,28看作是16张大小为28的三通道图片,如果我们要想只获得第1张和第三章图片,我们可以这样操作a.index_select(0,([0,2]))

第一个参数0表示操作数据a的第一个维度,也就是第几章图片,第二个参数表示获取第一维度的第一张还有第三张图片,注意第二个参数必须为tensor类型

这个还有一个参数没有使用就是index_select方法的第三个参数,这个还可以获取指定的索引数据。

masked_select</