华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
如图片所示,CANN的LayerNorm算子有三个输出,Pytorch的LayerNorm算子只需要一个输出Output_x,请问应该如何映射。目前是把Pytorch的算子映射成了相同结构的ONNX算子,但是CANN目前不支持ONNX的LayerNorm算子,看了说明文档,可以直接开发插件进行算子适配,但是输出的个数不同,应该怎么处理。
我要发帖子
如图片所示,CANN的LayerNorm算子有三个输出,Pytorch的LayerNorm算子只需要一个输出Output_x,请问应该如何映射。目前是把Pytorch的算子映射成了相同结构的ONNX算子,但是CANN目前不支持ONNX的LayerNorm算子,看了说明文档,可以直接开发插件进行算子适配,但是输出的个数不同,应该怎么处理。