INT64的onnx怎么转换成INT32的?
时间: 2024-05-20 08:15:52 浏览: 404
基于python的tensorrt int8 量化yolov5 onnx模型实现
5星 · 资源好评率100%
你好,对于你的问题,我可以为你提供一些参考信息。在ONNX转换中,可以使用Cast Op将INT64转换为INT32。具体操作是,在ONNX模型中添加Cast Op节点,并将该节点的输入数据类型设置为INT64并将输出数据类型设置为INT32。这样就可以将INT64类型的数据转换为INT32类型的数据。希望这能帮到你。
阅读全文