index_select()

可以快速上手的开发文档

#index_select()

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

函数形式:index_select(dim,index)参数:dim:表示从第几维挑选数据,类型为int值;index:表示..