{"id":205104,"date":"2025-05-29T12:12:23","date_gmt":"2025-05-29T04:12:23","guid":{"rendered":"https:\/\/server.hk\/cnblog\/205104\/"},"modified":"2025-05-29T12:12:23","modified_gmt":"2025-05-29T04:12:23","slug":"%e5%a6%82%e4%bd%95%e4%bd%bf%e7%94%a8-torch-onnx-export-%e5%af%bc%e5%87%ba%e7%9a%84-onnx-%e6%a8%a1%e5%9e%8b%e8%bf%9b%e8%a1%8c%e9%a2%84%e6%b5%8b%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/205104\/","title":{"rendered":"\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 ONNX \u6a21\u578b\u8fdb\u884c\u9884\u6d4b\uff1f"},"content":{"rendered":"

<\/b> <\/p>\n

\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 ONNX \u6a21\u578b\u8fdb\u884c\u9884\u6d4b\uff1f<\/h1>\n

\u4f60\u5728\u5b66\u4e60\u76f8\u5173\u7684\u77e5\u8bc6\u5417\uff1f\u672c\u6587\u300a\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 ONNX \u6a21\u578b\u8fdb\u884c\u9884\u6d4b\uff1f\u300b<\/span>\uff0c\u4e3b\u8981\u4ecb\u7ecd\u7684\u5185\u5bb9\u5c31\u6d89\u53ca\u5230<\/span>\uff0c\u5982\u679c\u4f60\u60f3\u63d0\u5347\u81ea\u5df1\u7684\u5f00\u53d1\u80fd\u529b\uff0c\u5c31\u4e0d\u8981\u9519\u8fc7\u8fd9\u7bc7\u6587\u7ae0\uff0c\u5927\u5bb6\u8981\u77e5\u9053\u7f16\u7a0b\u7406\u8bba\u57fa\u7840\u548c\u5b9e\u6218\u64cd\u4f5c\u90fd\u662f\u4e0d\u53ef\u6216\u7f3a\u7684\u54e6\uff01<\/p>\n

<\/p>\n

\u8c03\u7528 torch.onnx.export \u5bfc\u51fa\u7684 onnx \u6a21\u578b<\/strong><\/p>\n

\u672c\u6587\u65e8\u5728\u89e3\u7b54\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 onnx \u6a21\u578b\u3002<\/p>\n

\u95ee\u9898\uff1a<\/strong><\/p>\n

\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 onnx.pb \u6a21\u578b\u6587\u4ef6\uff1f<\/p>\n

\u89e3\u7b54\uff1a<\/strong><\/p>\n

pytorch \u6a21\u578b\u7684\u8f93\u5165\u4e3a tensor\uff0c\u800c onnx \u7684\u8f93\u5165\u4e3a numpy \u6570\u7ec4\u3002\u56e0\u6b64\uff0c\u5c06\u8f93\u5165\u6570\u636e\u4ece tensor \u8f6c\u6362\u4e3a numpy \u6570\u7ec4\u5373\u53ef\u89e3\u51b3\u95ee\u9898\u3002<\/p>\n

\u793a\u4f8b\u4ee3\u7801\uff1a<\/strong><\/p>\n

import onnxruntime\nimport numpy as np\n\nresnet_onnx = onnxruntime.InferenceSession('onnx.pb')\nx = np.ones((2, 2), dtype=np.float32)\ninputs = {resnet_onnx.get_inputs()[0].name: x}\nprint(resnet_onnx.run(None, inputs))<\/pre>\n
\u6ce8\u610f\uff1a<\/strong><\/p>\n
    \n
  • \u5c06 torch.tensor \u8f6c\u6362\u4e3a numpy \u6570\u7ec4\u3002<\/li>\n
  • \u4f7f\u7528 numpy \u6570\u7ec4\u4f5c\u4e3a onnx \u6a21\u578b\u7684\u8f93\u5165\u3002<\/li>\n
  • onnx \u6a21\u578b\u7684\u8f93\u51fa\u662f\u4e00\u4e2a\u5217\u8868\uff0c\u5305\u542b numpy \u6570\u7ec4\u3002<\/li>\n<\/ul>\n
    \u4eca\u5929\u5173\u4e8e\u300a\u5982\u4f55\u4f7f\u7528 torch.onnx.export \u5bfc\u51fa\u7684 ONNX \u6a21\u578b\u8fdb\u884c\u9884\u6d4b\uff1f\u300b\u7684\u5185\u5bb9\u5c31\u4ecb\u7ecd\u5230\u8fd9\u91cc\u4e86\uff0c\u662f\u4e0d\u662f\u5b66\u8d77\u6765\u4e00\u76ee\u4e86\u7136\uff01\u60f3\u8981\u4e86\u89e3\u66f4\u591a\u5173\u4e8e\u7684\u5185\u5bb9\u8bf7\u5173\u6ce8\u516c\u4f17\u53f7\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"
    \u5982\u4f55\u4f7f\u7528 torch.onnx....<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4925],"tags":[],"class_list":["post-205104","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/205104","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=205104"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/205104\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=205104"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=205104"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=205104"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}