模型里面含有ATan算子导致mindspore不能把ckpt模型转化为onnx

1 系统环境

硬件环境(Ascend/GPU/CPU): Ascend
MindSpore版本: mindspore=2.6.0
执行模式(PyNative/ Graph): 不限
Python版本: Python=3.9
操作系统平台: Linux

2 报错信息

2.1 问题描述

如果换成ASin算子,还是不行,如果用数学公式替换,误差太大,有没有别的等效办法?

2.2 报错信息

Traceback (most recent call last):
File “/mnt/c/Users/33635/Desktop/DWPCNNFusion/onnx.py”, line 41, in
main()
File “/mnt/c/Users/33635/Desktop/DWPCNNFusion/onnx.py”, line 37, in main
export(net, input_ir, input_vis, file_name=‘fusion_model.onnx’, file_format=‘ONNX’)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2071, in export
_export(net, file_name, file_format, *inputs, **kwargs)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2126, in _export
_save_onnx(net, file_name, *inputs, **kwargs)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2183, in _save_onnx
onnx_stream = _executor._get_func_graph_proto(net, graph_id)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/common/api.py”, line 2266, in _get_func_graph_proto
return self._graph_executor.get_func_graph_proto(exec_id, ir_type, incremental)
RuntimeError: Can not find key Asin in convert map. Exporting Asin operator is not yet supported.

3 根因分析

此处由用户填写~

4 解决方案

此处由用户填写~
包含文字方案和最终脚本代码 请将正确的脚本打包并上传附件