在使用torch.gather对数据进行收集的时候,为什么会有transpose算子参与
收藏回复举报
回复仅回复人和楼主可见
在使用torch.gather对数据进行收集的时候,为什么会有transpose算子参与
t('forum.solved') 已解决
发表于2025-11-12 09:42:31
0 查看
cke_201.png 对x利用gather操作后,看profile文件,产生了aclnnGather_TransposeAiCore_Transpose类型的算子,并且时延还很高,这是为什么?如何优化?

我要发帖子