问:可以将包含可变维度的input的pytorch 模型,导出成onnx吗?比如NLP模型,对可变长度字符串的输入。
答:可以,导出的时候你需要把对应的维度设置dynamic_axes, 请参考:https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html
问:可以将包含可变维度的input的pytorch 模型,导出成onnx吗?比如NLP模型,对可变长度字符串的输入。
答:可以,导出的时候你需要把对应的维度设置dynamic_axes, 请参考:https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html