Pytorch的LayerNorm算子如何映射到CANN的LayerNorm算子
收藏回复举报
Pytorch的LayerNorm算子如何映射到CANN的LayerNorm算子
t('forum.solved') 已解决
新人帖
发表于2023-04-19 16:47:48
0 查看

如图片所示,CANN的LayerNorm算子有三个输出,Pytorch的LayerNorm算子只需要一个输出Output_x,请问应该如何映射。目前是把Pytorch的算子映射成了相同结构的ONNX算子,但是CANN目前不支持ONNX的LayerNorm算子,看了说明文档,可以直接开发插件进行算子适配,但是输出的个数不同,应该怎么处理。

cke_194.png

我要发帖子