search

PaddlePaddle / Paddle2ONNX

ONNX Model Exporter for PaddlePaddle
Apache License 2.0
670 stars 157 forks source link

custom_ops映射使用问题 #1245

Open FrankyTang opened 1 month ago

FrankyTang commented 1 month ago

我在使用命令行进行put_along_axis映射，使用了custom_ops，但是onnx的输入和输出发现对不上 这个是我使用的命令行：paddle2onnx --model_dir . --model_filename paddle_put_along_axis.pdmodel --opset_version 16 --save_file put_along_axis.onnx --custom_ops '{"put_along_axis":"ScatterElements"}'

下面的图片可以看到输入的名称和变量都对不上，可能是我使用的不对，能否提供一个正确的使用教程

image

Zheng-Bicheng commented 1 month ago

@Jiang-Jia-Jun 这个参数还有在使用吗？

Zheng-Bicheng commented 1 month ago

这个好像没错呀，你这个输入变量名称原本应该是什么呢？