• H
    Add API and op for take_along_axis (#38396) · 3310f519
    huangxu96 提交于
    * add API and op for take_along_axis
    
    * fix compile dependency problem and add example code and doc
    
    * add unitest
    
    * delete some code for CI coverage
    
    * fix code style problem
    
    * fix as review
    3310f519
gather_scatter_kernel.cu 6.0 KB