pytorch gather函数

pytorch.gather  eg

四个输出 分别为 1 1 3 3 

if dim=0  3,4,1,2

 

posted @ 2023-01-13 14:45  夜未央111  阅读(9)  评论(0)  编辑  收藏  举报