#

index select

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

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

    作者:清晨的光明
    2020-10-08 14:31:14