#

index select

  • Pytorch中index_select() 函數(shù)的實(shí)現(xiàn)理解

    函數(shù)形式： index_select( dim, index ) 參數(shù)： dim：表示從第幾維挑選數(shù)據(jù)，類型為int值； index：表示從第一個(gè)參數(shù)維度中的哪個(gè)位置挑選數(shù)據(jù)，

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