2021
01-08
01-08
pytorch中index_select()的用法详解
pytorch中index_select()的用法index_select(input,dim,index)功能:在指定的维度dim上选取数据,不如选取某些行,列参数介绍第一个参数input是要索引查找的对象第二个参数dim是要查找的维度,因为通常情况下我们使用的都是二维张量,所以可以简单的记忆:0代表行,1代表列第三个参数index是你要索引的序列,它是一个tensor对象刚开始学习pytorch,遇到了index_select(),一开始不太明白几个参数的意思,后来查了一下资料,算是明...
继续阅读 >