y.view(-1,1) 转换成数组
gather(1, ) 按行取
self-Attention 与 transformer
最新推荐文章于 2021-02-04 15:53:29 发布
y.view(-1,1) 转换成数组
gather(1, ) 按行取