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

本文重点

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

index_select

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

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

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

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

masked_select</