index select

Pytorch中index_select() 函数的实现理解

函数形式： index_select( dim, index ) 参数： dim：表示从第几维挑选数据，类型为int值； index：表示从第一个参数维度中的哪个位置挑选数据，