{"id":10020970,"date":"2025-08-21T21:26:27","date_gmt":"2025-08-21T13:26:27","guid":{"rendered":"https:\/\/itmax.cn\/?p=10020970"},"modified":"2025-08-21T21:26:27","modified_gmt":"2025-08-21T13:26:27","slug":"%e3%80%90triton-%e6%95%99%e7%a8%8b%e3%80%91triton_language-interleave","status":"publish","type":"post","link":"https:\/\/itmax.cn\/?p=10020970","title":{"rendered":"\u3010Triton \u6559\u7a0b\u3011triton_language.interleave"},"content":{"rendered":"<p>Triton \u662f\u4e00\u79cd\u7528\u4e8e\u5e76\u884c\u7f16\u7a0b\u7684\u8bed\u8a00\u548c\u7f16\u8bd1\u5668\u3002\u5b83\u65e8\u5728\u63d0\u4f9b\u4e00\u4e2a\u57fa\u4e8e Python \u7684\u7f16\u7a0b\u73af\u5883\uff0c\u4ee5\u9ad8\u6548\u7f16\u5199\u81ea\u5b9a\u4e49 DNN \u8ba1\u7b97\u5185\u6838\uff0c\u5e76\u80fd\u591f\u5728\u73b0\u4ee3 GPU \u786c\u4ef6\u4e0a\u4ee5\u6700\u5927\u541e\u5410\u91cf\u8fd0\u884c\u3002<\/p>\n<p>\u66f4\u591a Triton \u4e2d\u6587\u6587\u6863\u53ef\u8bbf\u95ee \u2192<a href=\"https:\/\/link.segmentfault.com\/?enc=16bC0uBCpvqJOTdf4IzXng%3D%3D.OqxLvtW%2BR6iwtxz6J3luyZbW0VMvtsfm3ibBzk1WdWw%3D\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">https:\/\/triton.hyper.ai\/<\/a><\/p>\n<pre><code>triton.language.interleave(a, b)<\/code><\/pre>\n<p>\u6cbf\u7740\u6700\u540e 1 \u4e2a\u7ef4\u5ea6\u4ea4\u9519 2 \u4e2a\u5f20\u91cf\u7684\u503c\u3002\u8fd9 2 \u4e2a\u5f20\u91cf\u5fc5\u987b\u6709\u76f8\u540c\u5f62\u72b6\u3002\u7b49\u540c\u4e8e <em>tl.join(a, b).reshape(a.shape[-1:] + [2<strong><\/em><\/strong>a.shape[-1]])<em>* \u3002<\/em><\/p>\n<p><strong>\u53c2\u6570<\/strong>\uff1a<\/p>\n<ul>\n<li><strong>a<\/strong>\u00a0(<em>Tensor<\/em>) \u2013 \u7b2c 1 \u4e2a\u8f93\u5165\u5f20\u91cf\u3002<\/li>\n<li><strong>b<\/strong>\u00a0(<em>Tensor<\/em>) \u2013 \u7b2c 2 \u4e2a\u8f93\u5165\u5f20\u91cf\u3002<\/li>\n<\/ul>\n<p><a href=\"https:\/\/segmentfault.com\/a\/1190000047204509\">https:\/\/segmentfault.com\/a\/1190000047204509<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"<p>Triton \u662f\u4e00\u79cd\u7528\u4e8e\u5e76\u884c\u7f16\u7a0b\u7684\u8bed\u8a00\u548c\u7f16\u8bd1\u5668\u3002\u5b83\u65e8\u5728\u63d0\u4f9b\u4e00\u4e2a\u57fa\u4e8e Python \u7684\u7f16\u7a0b\u73af\u5883\uff0c\u4ee5\u9ad8\u6548\u7f16\u5199\u81ea\u5b9a\u4e49 [&hellip;]<\/p>\n","protected":false},"author":2,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_jetpack_memberships_contains_paid_content":false,"footnotes":""},"categories":[4],"tags":[],"class_list":["post-10020970","post","type-post","status-publish","format-standard","hentry","category-share"],"jetpack_sharing_enabled":true,"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020970","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/users\/2"}],"replies":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=10020970"}],"version-history":[{"count":0,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020970\/revisions"}],"wp:attachment":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=10020970"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=10020970"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=10020970"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}